Model 我加载一个经过训练的模型,损失相当大

Model 我加载一个经过训练的模型,损失相当大,model,pytorch,Model,Pytorch,我培训了一个翻译模型: 当步骤%1000==0时,我在验证时评估损失 如果损失很小,我就拯救国家 当我完成训练时,我加载模型,损失相当大 if step % 1000 ==0: model.eval() val_loss = eval(model,val_loader) print("epoch:",epoch,"step:",step,"loss:",loss,"val_loss:",val_loss) if val_

我培训了一个翻译模型:

  • 步骤%1000==0
    时,我在验证时评估损失
  • 如果损失很小,我就拯救国家
  • 当我完成训练时,我加载模型,损失相当大

        if step % 1000 ==0:
            model.eval()
            val_loss = eval(model,val_loader)
            print("epoch:",epoch,"step:",step,"loss:",loss,"val_loss:",val_loss)
    
            if val_loss.item() < best_loss:
                best_loss = val_loss.item()
                best_epoch = epoch
                torch.save(model.state_dict(),'best_1.m')
    

    如果看不到您的eval函数定义,就很难提供帮助。您是否在
    model.load\u state\u dict(torch.load('best\u 1.m'))
    之前重新实例化模型?如果不看您的eval函数定义,就很难提供帮助。您是否在
    model.load\u dict(torch.load('best\u 1.m'))之前重新实例化模型
    model.load_state_dict(torch.load('best_1.m'))
    model.eval()
    val_loss = eval(model,val_loader)
    print('val_loss:',val_loss)