# Conda activate pytorch15
# Note val_batch_size = batch_size to evaluate the model

This code is based on StarGANv2(https://github.com/clovaai/stargan-v2) and style-based-gan-pytorch(https://github.com/rosinality/style-based-gan-pytorch). Please install the dependencies from both projects.


# how to train
1> Downloading the pretrained StyleGAN from style-based-gan-pytorch, and move it into 'stylegan_model'. Downloading the data ('animal.zip') from the link(https://drive.google.com/drive/folders/1I-kyMSDENnpUrcJNUpwFRk98TsQh1ph1?usp=sharing), uncompressing and moving it into 'stage2/data'.

2> Setting '--resume_iter' to the which model (stage1) you expect to use 

3> python main.py --mode train --num_domains 149 --w_hpf 0  --lambda_reg 1 --lambda_sty 1 --lambda_ds 2 --lambda_cyc 1 --batch_size 8 --eval_every 500 --sample_every 1000 --save_every 5000  --total_iters 1000000  --resume_iter 20000 --kd --kd_checkpoint_dir ../stage1/result/checkpoint --train_img_dir data/animal/train       --val_img_dir data/animal/valid      --eval_dir result/eval                --result_dir  result         --checkpoint_dir result/checkpoint                    --sample_dir result/sample

