diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-09-24 00:21:17 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-09-24 00:21:17 -0700 |
| commit | d6d90bffee0bd74826e7700110819f6a35f62611 (patch) | |
| tree | bdf1a1d2591d986378ea4e79b81d88c7952c0374 | |
| parent | 75c16069b6fa395c2dfa740602d088764ff09c14 (diff) | |
update pool_size 0 for pix2pix
| -rw-r--r-- | README.md | 2 | ||||
| -rwxr-xr-x | scripts/train_pix2pix.sh | 2 |
2 files changed, 2 insertions, 2 deletions
@@ -87,7 +87,7 @@ bash ./datasets/download_pix2pix_dataset.sh facades - Train a model: ```bash #!./scripts/train_pix2pix.sh -python train.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction BtoA --lambda_A 100 --dataset_mode aligned --no_lsgan --norm batch +python train.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction BtoA --lambda_A 100 --dataset_mode aligned --no_lsgan --norm batch --pool_size 0 ``` - To view training results and loss plots, run `python -m visdom.server` and click the URL http://localhost:8097. To see more intermediate results, check out `./checkpoints/facades_pix2pix/web/index.html` - Test the model (`bash ./scripts/test_pix2pix.sh`): diff --git a/scripts/train_pix2pix.sh b/scripts/train_pix2pix.sh index ffe8dbf..098c520 100755 --- a/scripts/train_pix2pix.sh +++ b/scripts/train_pix2pix.sh @@ -1 +1 @@ -python train.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction BtoA --lambda_A 100 --dataset_mode aligned --no_lsgan --norm batch +python train.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction BtoA --lambda_A 100 --dataset_mode aligned --no_lsgan --norm batch --pool_size 0 |