pytorch 学习
文章平均质量分 79
源免
这个作者很懒,什么都没留下…
展开
-
关于torch版本不同,在使用crossentropy()时的一些问题
突然灵光一闪,我想看看两个torch环境是否一致,查了下,自己电脑torch版本是 1.12.1,服务器是1.7.0 ,我有点忐忑不会真是 torch版本问题把,于是我查了下 官网doc。查了下发现说是Cross_entropy的target参数得是 index形式,不能为one-hot。折腾了半天,查了很多资料,都说不能用 one hot,可是我自己试的又是可以的,真想qtmd。其标签必须为0~n-1,而且必须为1维的,如果设置标签为[nx1]的,则也会出现以上错误。看到这,真相rnmd,折腾了一晚。原创 2023-05-24 21:38:32 · 169 阅读 · 0 评论 -
ST-GCN 复现:遇到多GPU并行训练时使用nn.DataParallel遇到的nn.ParameterList()为空的问题
近期复现ST-GCN代码时,发现遇到一个问题,其开源代码使用了 nn.DataParallel() 进行多 GPU 并行训练,同时使用 nn.ParameterList() 来构建参数列表;经过确认,这是pytorch的bug,原作者是在torch1.2上开源的代码,本人使用的是torch1.7,会有这个问题。原创 2023-05-08 11:04:56 · 438 阅读 · 1 评论