Pytorch 模型测试 batch_size=1 时出现的问题

使用dataloader读取数据集

batch_size=1时,若不加model.eval(),则数据在传入模型时BN层会不起作用,导致预测错误

你可能感兴趣的:(python,深度学习)