zhenqifu / usuir Goto Github PK

首先十分感谢作者们杰出的工作，终于有首个基于pytorch实现的无监督水下图像增强算法了，且性能上接近有监督sota，我正在做水下图像增强领域和图像分割领域的一些工作，正好需要无监督的水下图像增强算法，感谢您

在跑您的代码的时候，我发现当batch size调到1以上时，反向传播时会报loss不是标量的错误

RuntimeError: grad can be implicitly created only for scalar outputs

原因是Color Loss的维度变成了(Batch,1,1,1)，我尝试简单的对Color Loss计算代码进行求均值来修正了这个error，请问您觉得这样合理吗？或者还有更好的解决方案吗？

# Color Loss
loss_3 = color_loss(j_out)
B, _, _, _ = loss_3.shape
loss_3 = torch.sum(loss_3, 0) / B