search

FadeoN / Distilling-Image-Dehazing-With-Heterogeneous-Task-Imitation

51 stars 8 forks source link

RuntimeError: The size of tensor a (552) must match the size of tensor b (550) at non-singleton dimension 3 #1

Closed Heibaizhu closed 4 years ago

Heibaizhu commented 4 years ago

Excellent work! I meet the problem above when I use the Student class to predict haze free image. It seems the upsampling module may meet dismatch problem when the input size is not a multiple of 4?

FadeoN commented 4 years ago

It should be fixed with the new commit. Please try again.

Heibaizhu commented 4 years ago

It works! Thank you!