深度学习时推理影响训练

深度学习时推理影响训练

在确保如下的情况下

model.eval()
with torch.no_grad:
	test()
model.train()

发现推理还是影响训练结果，最终发现问题出现在了dataloader的随机性问题。
因为我的代码需要dataloader的随机，然而默认的shuffle顺序可能和代码的运行时间有关，依据时间来生成随机数，所以出现影响。
给dataloader设置generator参数即可