人生中第一个自己完成的debug!!!激动!!分享给大家。
发现问题:在AttnGAN代码复现中出现以下问题。
出现原因定位到如下:
解决问题:将原本的channel_axis=0改成channel_axis=2
是one_map的通道数出现问题,往上翻找到one_map对象,one_map的形状第三位才表示通道数,因此将原本的channel_axis=0改成channel_axis=2.即第三位数值作为通道数。
人生中第一个自己完成的debug!!!激动!!分享给大家。
出现原因定位到如下:
是one_map的通道数出现问题,往上翻找到one_map对象,one_map的形状第三位才表示通道数,因此将原本的channel_axis=0改成channel_axis=2.即第三位数值作为通道数。