调试问题
爱科研的瞌睡虫
这个作者很懒,什么都没留下…
展开
-
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 · 6274 阅读 · 0 评论 -
PyCharm查看服务器CUDA及CUDNN的版本号
PyCharm查看服务器CUDA及CUDNN的版本号原创 2021-01-06 18:36:05 · 5223 阅读 · 0 评论