colab 跑 deformable-detr 记录:

!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

模型开始训练。

你可能感兴趣的:(深度学习,人工智能)