PETR
PETR copied to clipboard
请问代码不支持 torch2.2以上的版本吗?
依赖版本:
torch 2.3.0+cu118
torchaudio 2.3.0+cu118
torchvision 0.18.0+cu118
MMCV==2.2.0
执行命令:
./tools/dist_train.sh petrv2_vovnet_gridmask_p4_800x320_dn_centerview_benchmark_configs/petrv2_vovnet_gridmask_p4_800x320_dn_centerview_bs1_fp16_DP.py 1 --work-dir work_dirs/petrv2_vovnet_gridmask_p4_800x320_dn_centerview_bs1_fp16_DP/
报错:
Traceback (most recent call last):
File "/PETR/./tools/train.py", line 26, in
AssertionError: MMCV==2.2.0 is used but incompatible. Please install mmcv>=1.3.17, <=1.6.0.