pytorch
Y.K.Hou
西南交通大学测绘遥感信息系硕士在读
展开
-
3D点云基本网络模块(一):Spatial Transformer Networks(STN)
3D点云基本网络模块代码原创 2022-10-03 10:59:55 · 868 阅读 · 0 评论 -
关于loss不收敛的一些建议-loss问题汇总(不收敛、震荡、nan)
内容搬运自知乎:loss问题汇总(不收敛、震荡、nan)之前训练网络时,会先编写学习率随训练epoch的增加而逐渐减低的函数,然后选取一个相对较大的学习率(一般从e-2量级开始),选取一个epoch能够接受的batchsize,如果loss稳定下降较快,则开始训练.从未体验过学习率和batchsize搭配之难.最近新看了一篇论文ABCNN(有空再细讲),采用开源的tensorflow工程训练一下,效果很好,因工程需要,开始将其移植到pytorch框架下,移植完毕后,关于loss函数遇到不少问题,在此记录转载 2022-04-15 11:36:54 · 11096 阅读 · 1 评论 -
subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘, ‘-j‘, ‘12‘]‘ returned non-zero exit status 1
安装MinkowskiEngine编译的时候出现报错subprocess.CalledProcessError: Command ‘[‘ninja’, ‘-v’, ‘-j’, ‘12’]’returned non-zero exit status 1.后来查询发现是因为pytorch的版本过高,从1.11降到1.9之后,就可以编译成功了...原创 2022-03-14 19:40:54 · 1097 阅读 · 0 评论 -
Version of installed CUDA didn‘t match package该如何解决?
manjaro系统+3080显卡cuda11.1+conda虚拟环境(python3.8+pytorch1.7.1+cudatoolkit11.0.2)好不容易把包装好了,但是进入虚拟环境之后提示Version of installed CUDA didn’t match package这种情况下应该如何排查错误,寻找原因检测torch和驱动都是可以用的...原创 2021-03-20 21:46:53 · 749 阅读 · 1 评论 -
[linux]anaconda环境下载pytorch缓慢解决办法
[linux]anaconda环境下载pytorch缓慢解决办法我使用的是基于archlinux的manjaro系统,利用anaconda创建pytorch下载缓慢,科学上网无法解决,看了很多换国内源的方法,目前的换清华源的方法为:conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsin原创 2021-03-11 18:58:43 · 542 阅读 · 2 评论