Computing Resources

[rogershijin]

Could you give an idea of what computing resources you used to train your models and how long training took (along with how many epochs you trained for)?

[yu4u]

I'm using GTX 1080 / 1080Ti.
One epoch requires a few minutes.
Default training script runs for 60 epochs.

[AllenJac]

I have some questions about the train code, "train.py". When I run the code, all parameters are default, there is no mistake, but the code never stop. So I just use one image to train, and set "epoch=1", it still doesn't stop. Finally, I remove all my training images and set ("--image_dir", default=" ",), there is no path to train, it still can run without any mistakes. Do you know what's the reason and why it doesn't stop even train one image?

[yu4u]

I think I fixed this problem. #24
Could you try again after pulling the repository to adopt the PR?

[AllenJac]

Thanks for fixing it. But there is still something weird. I just use one image ( .jpg) from your training dataset to train, and set "epoch=2", other parameters are default, it still doesn't stop (Just keep "Epoch 1/2" and never stop), are there any thing I ignore?

[yu4u]

What is the training image size? If the image size is too small (< 64), training falls into an infinite loop...