search

NVlabs / stylegan3

Official PyTorch implementation of StyleGAN3
Other
6.28k stars 1.1k forks source link

can't training model #241

Open Pi-777 opened 12 months ago

Pi-777 commented 12 months ago

when I use python train.py xxxxx I got this report:

Setting up PyTorch plugin "upfirdn2d_plugin"... Done. Traceback (most recent call last):

xxxxxxxxxxxxx/training/networks_stylegan2.py", line 657, in forward y = x.reshape(G, -1, F, c, H, W) # [GnFcHW] Split minibatch N into n groups of size G, and channels C into F groups of size c. RuntimeError: shape '[4, -1, 1, 512, 4, 4]' is invalid for input of size 81920

did anyone meet this report like above and fix it?

jasuriy commented 2 months ago

@Pi-777 hi did you fix the issue? were you able to train the model with your own dataset?