I think you have same issue hero.
Are your feature maps in size [10, 208, 208]? In this case again you have multiplied by 10 so you need to use
outputs = outputs.view(10, 43264) #8028160

instead of:



 alex_gilabert:

—> 16 outputs = outputs.view(10, 432640) #8028160