![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 79
_胡不归
这个作者很懒,什么都没留下…
展开
-
设置CUDA_VISIBLE_DEVICES的方法
通过bash脚本设置CUDA_VISIBLE_DEVICESset CUDA_VISIBLE_DEVICES=0, 1, 2, 3source ~/anaconda3/bin/activate torchpython -c 'import torch; print(torch.__version__); print(torch.cuda.device_count)'原创 2020-09-15 00:13:14 · 59969 阅读 · 2 评论 -
torch.nn.Module中modules和children的区别
torch.nn.Module中modules()和children()的区别pytorch中一切网络结构一般都是继承自nn.Module类来实现的。nn.Module中存在两个相似的method:torch.nn.modules()和torch.nn.Modules.children()原创 2020-09-15 00:04:42 · 671 阅读 · 0 评论 -
如何设置随机种子保证pytorch代码的可复现性
在深度学习研究领域,论文结果的可复现性是一个很大的问题.遑论各种paper中的代码,有时候就是自己写的代码,都难以保证可复现性:即使使用同样的网络结构,同样的数据库,在同一台机器上训练,训练的结果都有差别.这一现象很大程度上是由于深度学习训练过程中的随机性造成的.网络参数的随机初始化正则化方法,例如dropout在训练中随机丢弃网络中的节点优化过程,例如SGD,RMSPorp或者Adam等...原创 2020-04-30 15:52:53 · 1858 阅读 · 0 评论