Skip to content

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

@Haruko386

Description

@Haruko386

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 ?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions