Hi, I use a pre-train model like below
    model= Net().to(device)  
    model.load_state_dict(torch.load("cnn_alex.pkl"))

and I set model. train() before training but It doesn’t converge. Can anyone help me?