From 2e75218b87b63421a2de91341252cd61884371db Mon Sep 17 00:00:00 2001 From: Shafeen Tejani Date: Sat, 7 Jan 2017 17:06:37 +0000 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 2808890..19da42e 100644 --- a/README.md +++ b/README.md @@ -30,9 +30,9 @@ The algorithm will run with the following settings: NUM_EPOCHS=5 # override with --epochs argument BATCH_SIZE=4 # override with --batch-size argument LEARNING_RATE = 1e-3 # override with --learning-rate argument -CONTENT_WEIGHT = 7.5 # override with --content-weight argument -STYLE_WEIGHT = 1e2 # override with --style-weight argument -TV_WEIGHT = 2e2 # override with --tv-weight argument +CONTENT_WEIGHT = 1 # override with --content-weight argument +STYLE_WEIGHT = 5 # override with --style-weight argument +TV_WEIGHT = 1e-6 # override with --tv-weight argument ``` To train the network using a GPU run with the `--use-gpu` flag.