김재형

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))
else:
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 y_chops:
y_chops = [[_y] for _y in y]
......