When does "collapse" happens?

[yh97yhyh]

Hello! I ran this code, but I had met "collapse" when 1epoch was over. (I edited code...) When does "collapse" happens? and How can I solve this problem?

[SamYuen101234]

Do you try some models pre-trained on ImageNet or VGG2?

[yh97yhyh]

No, How can I try these models?

[yh97yhyh]

Do you try some models pre-trained on ImageNet or VGG2?

Do you mean these models?

I just tried main.py and main2.py after put csv file, train and test dataset on folder. I didn't use these pre-trained models.

[SamYuen101234]

Do you set the parameter "pretrained" = True ? And what Model do you use. If you set pretrained = True, the models are pretrained on ImageNet/VGG2 are used.

[yh97yhyh]

Thanks! I'll give it a try!

[SamYuen101234]

closed

[SamYuen101234]

closed