分类：测试训练

2021
08-06

解决Pytorch在测试与训练过程中的验证结果不一致问题

引言今天在使用Pytorch导入此前保存的模型进行测试，在过程中发现输出的结果与验证结果差距甚大，经过排查后发现是forward与eval()顺序问题。现象此前的错误代码是input_cpu=torch.ones((1,2,160,160))target_cpu=torch.ones((1,2,160,160))target_gpu,input_gpu=target_cpu.cuda(),input_cpu.cuda()model.set_input_2(input_gpu,target_gpu)model.eval()model.forward()应该改为input_... 继续阅读 >