SKNet-2 input(x + v_Channel) bug 记录

1. SKNet输入要求x.shape=[2,3,256,256]

直接concat[input, V_chanel] ---------shape=[2,4,256,256]

报错显示:

 解决办法:conv(4,3,3,3)通过out_chanel=3强制转换为3通道

2.测试epoch=10,ssim=0.1, psnr=24.5 两个损失异常 未解决

SKNet在MIT的损失:

 SKNet-2在lolv1的损失

 

posted @ 2023-05-09 20:33  helloWorldhelloWorld  阅读(20)  评论(0)    收藏  举报