torch.save({'state_dict':model.state_dict()},'./Word_embdding_state_dict.pth')
model_dict = torch.load('./Word_embdding_state_dict.pth')
model.load_state_dict(model_dict['state_dict'])
model.eval()
这是保存和加载的代码
我尝试过训练完的时候立马保存加载,准确率是很高的。但是等到重新运行一遍的时候加载,就完全不对了,准确率特别低。
我一度认为是nn.Embedding函数生成的词向量的值每次都是随机的才导致这个问题。
实在不知道如何解决,求各位回答一下!!