uncbiag/SimpleClick

GPU setting

che1007 opened this issue · 3 comments

che1007 commented

Can I train your model on single GPU?
If the answer is 'Yes', what hyperparameter that I need to revise

qinliuliuqin commented

You just need to set "--ngpus=1" for training models on the branch v1.0.
I am now refactoring the code on the main branch, so don't train your models on the main branch for now.

  • 👍1
che1007 commented

Thanks for your help. The last two day I had tried to train base_448_sbd several times, and I found that the training loss went to rounding 0.31 after two or three epochs. But the training loss didn't keep going down after two or three epochs.
Expect the refactoring version from you.

qinliuliuqin commented

That's normal because we use normalized focal loss. Yes, I will refactor v1.0 soon because it's really messy.

  • 👍1