首页 > 其他 > 详细

Pytorch在使用过程中遇到的一些坑和小技巧

时间:2020-07-28 15:15:17      阅读:258      评论:0      收藏:0      [点我收藏+]

1. torch.backends.cudnn.deterministic = True 固定cuda的随机数种子,但是却有一个问题,在pytorch 1.5下加入这一行命令后,dilated Conv的计算速度会变慢数十倍,问题应该出在cudnn上,但是我也不知道到底是什么原因,github上有相关issue,见https://github.com/pytorch/pytorch/issues/28777

2. 在代码前加设torch.backends.cudnn.benchmark = True可以提升训练速度,注意,这只在训练过程中计算图是不变的情况下有效

Pytorch在使用过程中遇到的一些坑和小技巧

原文:https://www.cnblogs.com/ZAmateur/p/13390539.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!