summaryrefslogtreecommitdiff
path: root/README.md
diff options
context:
space:
mode:
authorjunyanz <junyanz@berkeley.edu>2017-04-22 08:15:48 -0700
committerjunyanz <junyanz@berkeley.edu>2017-04-22 08:15:48 -0700
commit6c347282993d2e2db91b376d3113efa3774c3a22 (patch)
tree75ede706fde3f61e73fbb9e7ea9ed6e97aff2a56 /README.md
parenta7917caaeaefe51db959b8f3ae50a20e726fbd93 (diff)
add dropout option for G
Diffstat (limited to 'README.md')
-rw-r--r--README.md2
1 files changed, 1 insertions, 1 deletions
diff --git a/README.md b/README.md
index 331ae54..4bb1595 100644
--- a/README.md
+++ b/README.md
@@ -81,7 +81,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 --align_data --no_lsgan
+python train.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction BtoA --lambda_A 100 --align_data --use_dropout --no_lsgan
```
- 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`):