Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Would you consider releasing the training code? #5

Open
noranart opened this issue Nov 23, 2019 · 1 comment
Open

Would you consider releasing the training code? #5

noranart opened this issue Nov 23, 2019 · 1 comment

Comments

@noranart
Copy link

noranart commented Nov 23, 2019

Hi, this is a great repo. By any chance, would it possible to also release the training code?

@wlhsiao
Copy link

wlhsiao commented Jan 8, 2020

Hi,
Training details are in our paper's supplementary file. Specifically, we adopted pix2pixHD's model architecture and training recipe to train our model, so our training code is the same as their train.py. The arguments we passed in to the script are:
python ./train.py \ --dataroot ./datasets/humanparsing \ --name humanparsing \ --label_feat \ --checkpoints_dir <PATH_TO_SAVE_MODEL> \ --label_dir <PATH_TO_SEGMENTATION_MAPS> \ --img_dir <PATH_TO_IMG> \ --resize_or_crop pad_and_resize \ --loadSize 256 \ --fineSize 256 \ --save_epoch_freq 100 \ --label_nc 18 \ --output_nc 8 \ --color_mode Lab \ --no_style_loss \ --no_recon_loss \ --gpu_ids 0,1,2,3,4,5,6,7 \ --batchSize 8

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants