bevfusion(multi task) 复现笔记
根据自己的需要下载完整的数据集或者mini数据集。把Map expansion中解压的三个文件夹放在map的文件夹下(map文件夹下就有了7个文件,四个图片文件和三个文件夹)。多卡训练,只需要把np的参数后的1改成自己相应的卡的数据即可。多卡评估,只需要把np的参数后的1改成自己相应的卡的数据即可。保存退出,并更新环境,让添加的环境变量生效。根据自己的机器的型号安装相应的cuda和torch包。运行之后会生成相应的.pkl文件。可以使用一下命令查看包的版本。不同的模型存在包的版本冲突,最好使用虚拟环境。
原创
2024-01-08 20:53:25 ·
1626 阅读 ·
2 评论