From 0e945892d9fd1dd1978563e0d32fdaecebfae6e0 Mon Sep 17 00:00:00 2001 From: "Steven H. Wang" Date: Fri, 31 Aug 2018 18:35:05 -0700 Subject: [PATCH] readme: Fix example command's `--model` option `models/models.py` is looking for {FFG,MVG}Model, not {FFG,MVG}. --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index e176508..48747cf 100644 --- a/README.md +++ b/README.md @@ -36,8 +36,8 @@ Currently only MNIST dataset are currently supported, and only fully connected l More details in [option_parser.py](https://github.com/wlwkgus/NoisyNaturalGradient/blob/master/option_parser.py) ### Train - $ python train.py --model=FFG --batch_size=100 --lr=1e-3 --dataset=MNIST - $ python train.py --model=MVG --batch_size=100 --lr=1e-2 --dataset=MNIST --n=60000 + $ python train.py --model=FFGModel --batch_size=100 --lr=1e-3 --dataset=MNIST + $ python train.py --model=MVGModel --batch_size=100 --lr=1e-2 --dataset=MNIST --n=60000 ### Visualize - To visualize intermediate results and loss plots, run `python -m visdom.server` and go to the URL http://localhost:8097 @@ -79,4 +79,4 @@ Visualization code(visualizer.py, utils.py) references to pytorch-CycleGAN-and-p ## Author -[Tony Kim](https://github.com/wlwkgus) \ No newline at end of file +[Tony Kim](https://github.com/wlwkgus)