search

SudeepDasari / one_shot_transformers

23 stars 6 forks source link

Training time #4

Closed Aeronnies closed 3 years ago

Aeronnies commented 3 years ago

Do I have to train 2000 epochs to achieve the performance of the paper? Training an epoch is very time-consuming for me.

SudeepDasari commented 3 years ago

Hey, I train for ~60K iterations on 2 GPUs. This can finish in about a day for me. This is certainly much less than 2000 epochs!