The device and trainning time?

[woaipichuli]

Thanks for sharing your code. Now, I'm trying to train this model on my computer to see if my device can support its trainning. But it seems like I can not find any output value. Maybe it has not trigger the situation when is_done==True yet. Could you tell me how many GPU or CPU you have use to train your model? And how long you trainning process has lasted?