wtjiang98 / PSGAN

PyTorch code for "PSGAN: Pose and Expression Robust Spatial-Aware GAN for Customizable Makeup Transfer" (CVPR 2020 Oral)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Error when training data for the model

umarshehu opened this issue · comments

When i run train.py, an error is shown " layers.append(SpectralNorm(nn.Conv2d(3, conv_dim, kernel_size=4, stride=2, padding=1)))
TypeError: init() takes 1 positional argument but 2 were given
from Net.py file. Seems there is a bug in the code? Thanks.
"