neuroailab / LocalAggregation-Pytorch

88 stars 17 forks source link

how much time you spend on training a LA model? #2

Open marvin521 opened 4 years ago

marvin521 commented 4 years ago

It has taken me several hours while one epoch of training has not finished yet.

chengxuz commented 4 years ago

Hi Marvin, using 2 Titan-Xp gpus, one epoch takes like 2500s to 3000s, so it's less than 1 hour. The whole model training takes usually less than 1 week.