NVlabs / stylegan3

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

checkpoint resume error #488

Closed Tongmengjun closed 11 months ago

Tongmengjun commented 11 months ago

When I use the command python train_stylegan2.py configs/gan/stylegan2/c10_style64.gin vitgan --mode=aug_both --aug=diffaug --lbd_r1=0.1 --no_lazy --halflife_k=1000 --penalty=bcr --use_warmup --use_nerf_proj --resume checkpoint_log to resume training, it raises an error: `RuntimeError: Error(s) in loading state_dict for Generator: Missing key(s) in state_dict: "layers.0.attn.to_qkv.module.weight", "layers.0.attn.to_out.0.module.weight", "layers.0.ff.net.0.module.weight", "layers.0.ff.net.3.module.weight", "layers.1.attn.to_qkv.module.weight", "layers.1.attn.to_out.0.module.weight", "layers.1.ff.net.0.module.weight", "layers.1.ff.net.3.module.weight", "layers.2.attn.to_qkv.module.weight", "layers.2.attn.to_out.0.module.weight", "layers.2.ff.net.0.module.weight", "layers.2.ff.net.3.module.weight", "layers.3.attn.to_qkv.module.weight", "layers.3.attn.to_out.0.module.weight", "layers.3.ff.net.0.module.weight", "layers.3.ff.net.3.module.weight". Does it mean that the network was not saved completely?