ModuleNotFoundError: No module named 'models.models'

[FrankMelon1]

Thanks for your great work firstly!

when I run "bash scripts/stage1/train_1.sh", the errors come as below:

raceback (most recent call last):
File "train_stage1.py", line 10, in
from models.models import create_model_parser, create_optimizer_parser, save_models, update_models
ModuleNotFoundError: No module named 'models.models'
root@notebook-tekxt-jg27l:/workspace/cpfs-data/C2F-FWN-main# bash scripts/stage1/train_1.sh
Traceback (most recent call last):
File "train_stage1.py", line 10, in
from models.models import create_model_parser, create_optimizer_parser, save_models, update_models
ModuleNotFoundError: No module named 'models.models'

I am looking forward to your reply, thanks!

[wswdx]

Sorry, it was my fault. Some files seemed to be missing in my last uploading. I've now added them in the updated version. You can clone the newest code repository to fix these errors.