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函数定义,就很难提供帮助。您是否在
之前重新实例化模型?如果不看您的eval函数定义,就很难提供帮助。您是否在model.load\u state\u dict(torch.load('best\u 1.m'))
?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)