Train time

[Mr-ChenSH]

Hello，The model training time is relatively long. How long have you been using it？

[zeshanalvi]

It's taking 185s/epochs with a batch size of 1024 and data size of 10K on a P100 GPU.