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

[xiaoxiongli]

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]

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