torch.save(model.state_dict(), ‘best_model.pth‘)

NoSuchKey

Guess you like

Origin blog.csdn.net/AdamCY888/article/details/131354334