如何使用python3读取CKPT文件,而使用python2_¼_保存该文件

pickle

torch.save(filename, model.state_dict())

model

model = Model() # construct a new model
model.load_state_dict(torch.load(filename))

np.savez

torch.load tell it to decode ASCII strings to Python3 strings

pytorch pickle

checkpoint = torch.load("xxx.ckpt")
with open("xxx.pkl", "wb") as outfile:
    pickle.dump(checkpointfile, outfile)

pkl_file = open("xxx.pkl", "rb")
data = pickle.load(pkl_file, encoding="latin1")
torch.save(data, "xxx.ckpt")