Model.train() with pre-trained weights makes results all 0 while model.eval() is fine

Hello, thanks for your attention to this matter.

I want to continue to train a model with its pre-trained weights. When I evaluate this pre-trained model with model.eval(), everything is fine and the model will generate some reasonable results, but when I want to further train this model and set the mode with model.train(), the problem will occur. During the forward loop, all generated results will be zero after switching to model.train() statement (batchsize=1).

Any ideas about why this happens?

Thanks for your answer!