pytorch写程序,中文坐标轴无法显示出来
在前面已经定义过显示中文的语句了
保存整个模型:
torch.save(model_object, 'model.pt')
加载整个模型:
model = torch.load('model.pt')