
pytorch
pytorch记录
北京纯牛奶
做个知识点记录的工具而已,其他绕道!
展开
-
查看pytorch版本,本机和环境版本不对应,启动慢
pythonimport torchtorch.version.cuda以上就是你在该环境下安装的cuda版本,有的时候环境的版本和本机的版本不对应,搞得我每次运行起来出,启动时候特别慢,而且汇报版本错误。具体错误忘了截图,抱歉如果出现了这个问题,可以直接pytorch官网安装本地版本,直接在这个环境下面安装就可以,然后下面的10.0的版本就没了。...原创 2021-11-27 17:50:09 · 740 阅读 · 0 评论 -
Pytorch随机种子设置或固定
最近在做比赛的时候,遇到了一个最好结果,但是之后无论怎样都复现不出来最好结果了。猜测是不是跟Pytorch中的随机种子有关。训练过程在训练过程中,若相同的数据数据集,相同的训练集、测试集划分方式,相同的权重初始化,但是每次训练结果不同,可能有以下几个原因:Dropout的存在 Pytorch、Python、Numpy中的随机种子没有固定 数据预处理、增强方式采用了概率,若没有设置固...转载 2020-04-20 22:08:22 · 6689 阅读 · 3 评论 -
pytorch中残差网络resnet的源代码的解读
pytorch的resnet模块在torchvision的models中。里面可以选择的resnet类型有:_all_列表的每一个resnet都提供了实现的函数:def resnet18(pretrained=False, progress=True, **kwargs):“”“Constructs a ResNet-18 model.Args:pretrained (bool): ...转载 2020-03-18 20:02:02 · 1885 阅读 · 2 评论