CaptainJi

train.py文件save_network方法改成这样 ```python def save_network(network, epoch_label): save_filename = 'net_%s.pth'% epoch_label save_path = os.path.join('./model',name,save_filename) # torch.save(network.cpu().state_dict(), save_path) # 上面注释的部分改成下面的 torch.save(network, save_path) if torch.cuda.is_available(): network.cuda(gpu_ids[0]) ``` 然后新建一个py文件，内容如下（其中输入模型和输出模型路径改成自己的）： ```python import torch import torch.nn import...