Problems with importing timm/albumentations

[taijizhao]

Congratulations on winning the competition and thank you very much for sharing the awesome repository!
I am trying to follow your work and replicating the convnext training process.
The datasets preparation went fine but I got stuck in running
sh ./_train_script_auto_generated.sh
and got the error:
ImportError: src/pytorch-image-models/projects/rsna/exps/exp7_fully_reproduce.py doesn't contains class named 'Exp'
It seems that exp7_fully_reproduce.py cannot import timm and albumentations correctly.

I did not use the provided docker but manually pip installed the packages following the docker file.
should I pip install -e the pytorch-image-models and albumentations in the src folder?
Thank you very much!

[dangnh0611]

It seems that the Python interpreter can't find timm and albumentations package/module which located in src/pytorch-image-models. You can try:

export PYTHONPATH=$(pwd)/src/pytorch-image-models/:$PYTHONPATH

before running the train command. Or, add it directly to train command like:

PYTHONPATH=$(pwd)/src/pytorch-image-models/:$PYTHONPATH  python3 train_exp.py

Of course, your method pip install -e should also work as well
Hope it helps !

[taijizhao]

Thank you very much for the reply!
Problem solved.
Have a nice day!