# 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'

2> python main.py --mode train --num_domains 2 --w_hpf 0  --lambda_reg 1 --lambda_sty 1 --lambda_ds 2 --lambda_cyc 1 --batch_size 2 --val_batch_size 2 --sample_every 1000 --eval_every 2000 --save_every 5000 --total_iters 20000 \
               --train_img_dir data/useless/train \
               --val_img_dir data/useless/test \
               --result_dir  result \
	       --eval_dir result/eval \
 	       --checkpoint_dir result/checkpoint \
               --sample_dir result/sample



