今天训练一个小模型,发现训练时候batchsize设为16,训练过程中test出结果是100%;但是加载模型测试的时候,结果就不对了。。。尤其是batchsize设为2/4/8的时候。后来发现是自己蠢啊
漏了一句
model.eval()
今天训练一个小模型,发现训练时候batchsize设为16,训练过程中test出结果是100%;但是加载模型测试的时候,结果就不对了。。。尤其是batchsize设为2/4/8的时候。后来发现是自己蠢啊
漏了一句
model.eval()