2021
07-26
07-26
pytorch预测之解决多次预测结果不一致问题
为什么多次预测结果不一致1、检查是否在每次预测前使用model.eval()或者是withtorch.no_grad():for...推荐下面的方法,上面的的方法计算梯度,但是并不反向传播,下面的方法既不计算梯度,也不反向传播,速度更快。2、检查是否取消了所有的dropout3、设置随机种子defsetup_seed(seed):np.random.seed(seed)random.seed(seed)torch.manual_seed(seed)#cputorch.cuda.manual_seed_all(seed)#并行gputor...
继续阅读 >