Charlotte_DL的博客

transforms.ToTensor()transforms.ToTensor() 可以将 PIL.Image/numpy.ndarray 数据进转化为torch.FloadTensor，并归一化到[0, 1.0]：取值范围为[0, 255]的PIL.Image，转换成形状为[C, H, W]，取值范围是[0, 1.0]的torch.FloadTensor；形状为[H, W, C]的numpy.ndarray，转换成形状为[C, H, W]，取值范围是[0, 1.0]的torch.FloadTen

原文出处： https://ptorch.com/news/198.html最近pytorh更新到0.4.1版本，真的很多小伙伴反馈有代码在版本升级后不能运行；需要降级pytorch版本，所以我整理一下降级方法，方便大家以后查阅，节约各位大佬宝贵的时间！一、Conda降级pytorch版本或安装指定版本pytorch如果你是使用conda包管理，你可以很容易实现版本降级，你只需要指定版...

# 保存和加载整个模型torch.save(model_object, 'model.pkl')model = torch.load('model.pkl')# 仅保存和加载模型参数(推荐使用)torch.save(model_object.state_dict(), 'params.pkl')model_object.load_state_dict(torch.load('...