当前位置:   article > 正文

Stratified Transformer复现和调试运行记录,训练S3DIS数据集(点云语义分割)

stratified transformer

环境:

ubuntu20.04,gcc和g++都是7.5,cuda是11.3,显卡3090

创建虚拟环境

python=3.7,离线安装torch=1.8.0,torchision=0.9.0 (torch 不要超过1.12.1我的编译时报错)  安装torch=1.8.0(兼容性稍微好点)是因为torch.points3d这些库的要求,也可以使用作者用的torch1.7.1,cuda10.1,我的torch.points3d编译又报错了,哭

torch和torchision按好后直接怼图中红色的框,按顺序,编译torch_sparse和point3d估计用了快一个小时,在安装没有安装的库h5py==3.2.1,SharedArray==3.2.1,最后,注意不要随意安装numpy==1.19.5,可能会报错numpy.core什么的,这个还挺难的,万一出现了用conda和pip卸载conda uninstall numpy 和pip install numpy,中间使用conda clean -p 清理包,再用conda install numpy,这是一个可借鉴方法,不是百试百灵。

最后,祝你好运!!!

 11.20 补充一下在老师那儿跑的结果,S3DIS,在区域5上,3090训练完成时间40小时左右,best Iou=67.2,应该是平均Iou ,接下来,就是按部就班地跑6折交叉验证了!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/101567
推荐阅读
相关标签
  

闽ICP备14008679号