search

AAboys / MobileFormer

Code and models for mobile-former
119 stars 18 forks source link

Hyper-parameters for training #1

Closed YehLi closed 2 years ago

YehLi commented 2 years ago

Thanks for your great job. Do you plan to release the training code?

AAboys commented 2 years ago

The training command is added. Default train.py in timm is used.

YehLi commented 2 years ago

Thanks for your reply, do you use the same mixup, aa, reprob for other Mobile-Former models?

AAboys commented 2 years ago

it may change for smaller models, which model do you looking for?

wangwuwei commented 2 years ago

it may change for smaller models, which model do you looking for?

could you please provide the training commands of other models such as 96m?

AAboys commented 2 years ago

training setup for 96m is added.