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

环境:

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折交叉验证了!

  • 9
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 64
    评论
评论 64
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值