duanyiqun / duanyiqun/DiffusionDepth

No module named 'mmcv._ext' and ImportError: cannot import name 'MultiScaleDeformableAttention' from 'mmcv.cnn.bricks.transformer

Open
#54 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
341
Forks
23
PR merge metrics
No merged PRs in 30d

Description

I have pytorch2.0.1+cu118
mmcv-full=1.4.0
mmdet=2.14.0

when I try to start training with command
`python main.py --dir_data /root/DiffusionDepth/Dataset/nyudepthv2 --data_name NYU --split_json ../data_json/nyu.json \
--patch_height 352 --patch_width 906 --gpus 0 --loss 1.0*L1+1.0*L2+1.0*DDIM --epochs 30 \
--batch_size 8 --max_depth 88.0 --save NAME_TO_SAVE \
--model_name Diffusion_DCbase_ --backbone_module swin --backbone_name swin_large_naive_l4w722422k --head_specify DDIMDepthEstimate_Swin_ADDHAHI`

I got `No module named 'mmcv._ext' `or `ImportError: cannot import name 'MultiScaleDeformableAttention' from 'mmcv.cnn.bricks.transformer' (/root/anaconda3/envs/DL/lib/python3.8/site-packages/mmcv/cnn/bricks/transformer.py)`

Any suggestion ?

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.