pytorch问题解决汇总

1.训练模型时候报错

pytorch AttributeError: 'SRResNet' object has no attribute 'module'

经查，这是由多卡训练代码改成单卡训练情况下，保存模型时发生的错误，需要将原代码中

torch.save({
            'epoch': epoch,
            'model': model.module.state_dict(),
            'optimizer': optimizer.state_dict()
        }, 'results/checkpoint_srresnet.pth')

'model': model.module.state_dict(),的module.删除，修改后为

torch.save({
            'epoch': epoch,
            'model': model.state_dict(),
            'optimizer': optimizer.state_dict()
        }, 'results/checkpoint_srresnet.pth')