search

SerialLain3170 / adeleine

Automatic line art colorization using various types of hint or without hint
Other
355 stars 47 forks source link

A little mistake in pix2pix and pix2pixHD code #9

Closed GiantKevin closed 4 years ago

GiantKevin commented 4 years ago

https://github.com/SerialLain3170/Colorization/blob/66915f3875720f6223a51efaee358f3c3e315bef/nohint_pix2pixHD/model.py#L117

if norm == "bn": modules.append(nn.BatchNorm2d(out_ch)) elif norm == "in": modules.append(**nn.BatchNorm2d(out_ch)**) nn.BatchNorm2d(out_ch) should be nn.InstanceNorm2d(out_ch).

Anyway, this project is great, thanks for your sharing.

SerialLain3170 commented 4 years ago

Thanks, I had not noticed the bug. I'll fix it.