ModuleNotFoundError: No module named 'models.swin_transformer_3D'

[huangchieh]

Dear authors,

Thank you for making this code open-sourced. I followed the changes of #19 and created the training data.

Then I tried to run the vessel_graph example using the following command,

python train.py --config configs/synth_3D.yaml --cuda_visible_device 0  --nproc_per_node 1

and meet this error:

Traceback (most recent call last):
  File "train.py", line 16, in <module>
    from models import build_model
  File "/Github/relationformer/models/__init__.py", line 3, in <module>
    from .swin_transformer_3D import build_swin_transformer
ModuleNotFoundError: No module named 'models.swin_transformer_3D'

It seems the file swin_transformer_3D.py are missing:

>.../relationformer/models
└─▪ pwd
Github/relationformer/models
>../relationformer/models
└─▪ ls
__init__.py  __pycache__  deformable_transformer_3D.py  detr_transformer_3D.py  matcher.py  ops  position_encoding.py  relationformer.py  seresnet.py

Could you please give me some suggestions?

Thank you.