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


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可以提升训练速度,注意,这只在训练过程中计算图是不变的情况下有效


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM