ESDR 사용시 --chop 옵션을 사용할 수 없는 문제 해결

- https://github.com/sanghyun-son/EDSR-PyTorch/issues/184#issuecomment-502650995

@@ -132,7 +132,8 @@ class Model(nn.Module):
                         y_chop.extend(_y.chunk(n_GPUs, dim=0))
                         y_chop.extend(_y.chunk(n_GPUs, dim=0))
         else:
         else:
             for p in zip(*x_chops):
             for p in zip(*x_chops):
-                y = self.forward_chop(*p, shave=shave, min_size=min_size)
+                p1 = [p[0].unsqueeze(0)]
+                y = self.forward_chop(*p1, shave=shave, min_size=min_size)
                 if not isinstance(y, list): y = [y]
                 if not isinstance(y, list): y = [y]
                 if not y_chops:
                 if not y_chops:
                     y_chops = [[_y] for _y in y]
                     y_chops = [[_y] for _y in y]