Update README.md

HeDeYu · Jan 7, 2017 · 2e75218 · 2e75218
1 parent 04dd8a3
commit 2e75218
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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.