Only can train 1 epoch?

Question

Only can train 1 epoch?

tnaderi opened this issue 4 years ago · 3 comments

Hi YvanYin,
Thank you for sharing your code. I can not train the model. When it goes over small NYUV2 one time it becomes idle. Does not do anything. How can I solve this?
Best,
Taher Naderi

Answer 1 · 2021-02-06T01:41:01.000Z

It seems to me that it can initialize randomly, but it can not initialized from a pretrained model.

Answer 2 · 2021-02-15T19:03:09.000Z

I do not get your reported results in ablation study. Could you please tell me what setting you are using?

Answer 3 · 2021-03-19T02:00:03.000Z

Please check the path of your pretrained weight.
When you train the model on small NYU, you can set the LR to 0.01.