“reflection_pad2d_out_template“ not implemented for ‘Byte‘

pytorch
问题：“reflection_pad2d_out_template” not implemented for ‘Byte’
解决：HR_tor_img = torch.from_numpy(HR_np_img.copy()).float()

涉及的输入数据要加上.float()
网络改成net=net.cuda()

版权声明：本文为mjx792原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。