autonomousvision / stylegan-xl

[SIGGRAPH'22] StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets
MIT License
961 stars 113 forks source link

Error when training stylegan2 #97

Open mingqiJ opened 1 year ago

mingqiJ commented 1 year ago

When I use this command to train stylegan2:python3 train.py --outdir=./out --cfg=stylegan2 --cond=True --data=./dataset/cifar10.zip --gpus=4 --batch=128 --mirror=True --batch-gpu 32 --cbase 16734 --cmax 256 --kimg 25000 --snap 200 --syn_layers 7 --head_layers 4 It takes errors: TypeError: init() got an unexpected keyword argument 'num_layers'

WeChatab2450b57785acea5d3a187e241820c9

But I can successfully train stylegan3-r and fastgan.

It is so weird. How to solve these problem/ Has anyone successfully run these two models?