Triority's blog

在torch1.6版本及其以后，torch.save函数使用了一种新的文件格式。torch.load任然保持着对旧版本的兼容，如果高版本环境下想要保存低版本兼容的模型文件格式，可以使用_use_new_zipfile_serialization=False参数设定 1234The 1.6 release of PyTorch switched torch.save to use a new zipfile-based file format. torch.load still retains the ability to load files in the old format. If for any reason you want torch.save to use the old format, pass the kwarg _use_new_zipfile_serialization=False. 但是如果低版本使用高版本模型文件就会报错，因此写了一个转换程序批量转换所有模型文件。 程序也在我的photo_data_maker仓库里，方便训练完直接调用 12345678 ...