Where is train.py

[Lizhuoling]

I do not see the file "train.py" for start training.

[wsy-yjys]

where is the train.py？

[kevinz8866]

Hi,

Check this out. This fork fixes all the import issues. https://github.com/kevinz8866/MobileFormer

[lhl12395]

@kevinz8866 Hi，did you encounter RuntimeError: Unknown model (mobile_former_294m) in timm? and if so, can you share some experience？Please

[kevinz8866]

@lhl12395 I can't recall that issue. Chechout my fork that fixes import issues or provide your steps to reproduce the issue so that I could take a closer look. Cheers.

[lhl12395]

@kevinz8866 Thanks for your reply. I resolve this problem just now. I forgot import mobile_former in _init.py. Have a good day. Cheers.

[wsy-yjys]

ok, thankyou

[syswyl]

@lhl12395

Hello, are you following

timm==0.6.12 torch==1.12.0+cu113 torchvision==0.13.0+cu113

environment configured?

I noticed the function create_scheduler_v2 in the code, but it seems that timm officially added this function only in the 0.8 dev version. When I upgraded the timm version, I got the _Unknown model (mobile_former_294m) in timm_ issue you mentioned. In which __init.py file did you add mobile_former?

[lhl12395]

Hello，You should execute the following code： git clone https://github.com/rwightman/pytorch-image-models cd pytorch-image-models && pip install -e .

This makes the timm editable.

[syswyl]

pip install -e . Thanks! I'll try it~