search

williamyang1991 / DualStyleGAN

[CVPR 2022] Pastiche Master: Exemplar-Based High-Resolution Portrait Style Transfer
Other
1.64k stars 254 forks source link

How do '(3) Training DualStyleGAN' on cpu #103

Open Ohjunghh opened 6 months ago

Ohjunghh commented 6 months ago

The code is no longer working at <Step 2: Fine-tune StyleGAN. Fine-tune StyleGAN in distributed settings: >

I do : finetune-stylegan.py line 279 - device ='cuda' -> 'cpu'

and

python finetune_stylegan.py --iter 100 --batch 32 --ckpt ./checkpoint/stylegan2-ffhq-config-f.pt --style cartoon --augment ./data/cartoon/lmdb/

But, There's nothing working in the terminal. I already did github.com/williamyang1991/DualStyleGAN/issues/92