[记录]-.pytorch转为.pth

import torch

# 加载原始模型
original_model = torch.load('network-chairs-things.pytorch')
# 指定要保存的文件名
save_path = 'network-chairs-things.pth'

# 保存原始模型为.pth文件
torch.save(original_model, save_path)


你可能感兴趣的:(深度学习,pytorch,人工智能)