机器学习
关注
分享
扫一扫
iRonMan丶
这个作者很懒，什么都没留下…
展开
-
linux怎么看正在运行的文件代码在哪个文件夹
https://blog.51cto.com/15023263/2558845原创 2021-02-04 20:59:15 · 844 阅读 · 1 评论 -
记录一个pytorch小bug：RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
This happens also in the windows port of PyTorch, the only way to overcome this when using (in my case) large CNN’s is to use:torch.backends.cudnn.enabled=False.原创 2019-03-22 10:48:34 · 4811 阅读 · 2 评论 -
the tricks of using pytorch
1.不要使用 +=、-=等连符，这样会改变原始张量的值，使在反向传播计算梯度时，出错。原创 2019-05-04 19:36:33 · 121 阅读 · 1 评论