search

nrupatunga / goturn-pytorch

GOTURN tracking implemented in PyTorch
MIT License
84 stars 13 forks source link

How many epochs did you train? #6

Closed lee-man closed 4 years ago

lee-man commented 4 years ago

Hi,

Thanks for your excellent code. I just wanna check how many epochs did you train? 3 epochs? I am trying to train the model myself and it seems like it will take about 2 days for a single epoch.

Thanks!

nrupatunga commented 4 years ago

Hi,

I trained on 1080Ti, It should train within a day. What is data loader speed, how many iters/sec, does it show in the progress bar.

lee-man commented 4 years ago

It took about 5 seconds per iteration... I will check the setting. Thanks for your reply.

nrupatunga commented 4 years ago

This figure also gives you an idea. (look at the x-axis)

https://github.com/nrupatunga/goturn-pytorch/blob/master/doc/images/loss.png

Also, data on SSD drive might speedup, though I remember my data was on normal HDD.