!GPUS_PER_NODE=1 ./tools/run_dist_launch.sh 1 ./configs/r50_deformable_detr.sh
报错:
cannot import name '_NewEmptyTensorOp' from 'torchvision.ops.misc'
经issue查阅:试用意见:
这是由于在torchvision 0.10版本的util/misc.py中的一个错误。代码只检查前3个字符,所以它认为你是在0.1版本上,并试图导入_NewEmptyTensorOp。 我通过删除util/misc.py中的第30-59和490-501行来修复它。
问题解决了!
问题解决,开始报另一个错:
ModuleNotFoundError: No module named 'MultiScaleDeformableAttention'
经issue查询,试用意见:
you should compile CUDA operators
cd ./models/ops
sh ./make.sh
命令:
%cd models/ops
!sh /content/Deformable-DETR/models/ops/make.sh
运行结果:
Finished processing dependencies for MultiScaleDeformableAttention==1.0
模型开始训练。