pytorch-save and load models


1.

torch.save(model_name.state_dict(), name_to_save)


net = ModelClass(args)

net.load_state_dict(torch.load(PATH))


2.

torch.save(model, PATH)


net = torch.load(PATH)


你可能感兴趣的:(PyTorch)