it seems have a bug when I train a x2(scale=2) model?

NJU-Jet / SR_Mobile_Quantization

Winner solution of mobile AI (CVPRW 2021).

148 stars 41 forks source link

Open xiaoxiongli opened 3 years ago

xiaoxiongli commented 3 years ago

Hi Dear @NJU-Jet:

I wanna to train a model that scale=2. (as you know, the defaut scale is 3) it seems have a bug here:

you can see that this [inp, inp, inp, inp, inp, inp, inp, inp, inp] is for scale=3, so how to fix it if I want to train a x2 model?

can we fix it this way: [inp, inp, inp, inp, inp, inp, inp, inp, inp] --> [inp] x scale x scale ?

thank you very much!

NJU-Jet commented 3 years ago

You are right and this is a bug. The inp should just repeat (scale**2) times.