转自
需要更正的地方:
1.
learning_rate = 1e-3 # 学习率
学习率调的低一点比较好,不然容易出现正确率都是10%的情况
2.
测试模型阶段:
原先的代码为:
img = Variable(img, volatile=True).cuda()
label = Variable
转自
需要更正的地方:
1.
learning_rate = 1e-3 # 学习率
学习率调的低一点比较好,不然容易出现正确率都是10%的情况
2.
测试模型阶段:
原先的代码为:
img = Variable(img, volatile=True).cuda()
label = Variable