From 43514362a5d82b8e488ccc237dba91e2a7e930da Mon Sep 17 00:00:00 2001 From: Yong Date: Tue, 14 Aug 2018 19:45:30 +0800 Subject: [PATCH] update README --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 92a75db..bbe8f91 100644 --- a/README.md +++ b/README.md @@ -50,11 +50,11 @@ In our paper, to sample different images, we train our model on four datasets, r ## Training - Train LCCGAN on Oxford-102 Flowers dataset. ``` -python train.py --dataset flowers --dataroot your_images_folder --batchSize 64 --imageSize 64 --cuda +python trainer.py --dataset flowers --dataroot your_images_folder --batchSize 64 --imageSize 64 --cuda ``` - If you want to train the model on Large-scale CelebFaces Attributes (CelebA), Large-scale Scene Understanding (LSUN) or your own dataset. Just replace the hyperparameter like these: ``` -python train.py --dataset name_o_dataset --dataroot path_of_dataset +python trainer.py --dataset name_o_dataset --dataroot path_of_dataset ``` ## Citation