![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
调试问题
爱科研的瞌睡虫
这个作者很懒,什么都没留下…
展开
-
ReLu函数中inplace参数的理解
ReLu函数中inplace参数的理解x = torch.nn.functional.relu(v(x), inplace=True)inplace = True是指原地进行操作,操作完成后覆盖原来的变量。例如:上面的代码中的意思v(x)进行relu之后赋值给x优点:节省内存缺点:进行梯度回归的时候传回失败,原来的变量被覆盖了,找不到原来的变量。...原创 2021-01-26 19:04:13 · 6044 阅读 · 0 评论 -
PyCharm查看服务器CUDA及CUDNN的版本号
PyCharm查看服务器CUDA及CUDNN的版本号原创 2021-01-06 18:36:05 · 4953 阅读 · 0 评论