StyleSwin icon indicating copy to clipboard operation
StyleSwin copied to clipboard

How to finetune?

Open GinIsTheAnswer opened this issue 2 years ago • 2 comments

Sorry if this is a silly question, but I wanted ask if you could provide and example of how to fine-tune one of your existing models to a new dataset?

An example from StyleGan3 # Fine-tune StyleGAN3-R for MetFaces-U using 1 GPU, starting from the pre-trained FFHQ-U pickle. python train.py --outdir=~/training-runs --cfg=stylegan3-r --data=~/datasets/metfacesu-1024x1024.zip \ --gpus=8 --batch=32 --gamma=6.6 --mirror=1 --kimg=5000 --snap=5 \ --resume=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-ffhqu-1024x1024.pkl

GinIsTheAnswer avatar Jul 15 '22 09:07 GinIsTheAnswer

You supply the path to the checkpoint via --ckpt, ie. --ckpt samples/000001.pt. Also don't forget to supply original hyperparams, like image size, style dim, if we use SN etc...

TheGullahanMaster avatar Jul 16 '22 06:07 TheGullahanMaster

You supply the path to the checkpoint via --ckpt, ie. --ckpt samples/000001.pt. Also don't forget to supply original hyperparams, like image size, style dim, if we use SN etc...

Thanks, that's right.

ForeverFancy avatar Jul 18 '22 12:07 ForeverFancy