NJU-Jet / SR_Mobile_Quantization

Winner solution of mobile AI (CVPRW 2021).
148 stars 41 forks source link

During QAT training for scale = 2, do I need change below "3" to "2" ? #7

Open xiaoxiongli opened 3 years ago

xiaoxiongli commented 3 years ago

I have already trained a model that scale = 2, and after that I wanna to do QAT training for it(scale = 2) using command "python train.py --opt options/train/base7_qat.yaml --name base7_D4C28_bs16ps64_lr12-3_qat_x2 --scale 2 --bs 16 --ps 64 --lr 1e-3 --gpu_ids 1 --qat --qat_path experiment/ base7_D4C28_bs16ps64_lr12-3_x2/best_status", do I need change below red box "3" to "2" ?

image

Do we need change like below?

image

thank you very much!

NJU-Jet commented 3 years ago

Yes, and thanks for finding this bug.