Skip to content
This repository was archived by the owner on Jan 2, 2021. It is now read-only.

Commit f68f04f

Browse files
committed
Improve instructions to train custom models so a new file is output and existing one is not loaded. Use --model parameter!
1 parent 203917d commit f68f04f

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

README.rst

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,16 +58,19 @@ Pre-trained models are provided in the GitHub releases. Training your own is a
5858
rm -f ne4x*.pkl.bz2
5959
6060
# Pre-train the model using perceptual loss from paper [1] below.
61-
python3.4 enhance.py --train "data/*.jpg" --scales=2 --epochs=50 \
61+
python3.4 enhance.py --train "data/*.jpg" --model custom --scales=2 --epochs=50 \
6262
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
6363
--generator-blocks=4 --generator-filters=64
6464
6565
# Train the model using an adversarial setup based on [4] below.
66-
python3.4 enhance.py --train "data/*.jpg" --scales=2 --epochs=250 \
66+
python3.4 enhance.py --train "data/*.jpg" --model custom --scales=2 --epochs=250 \
6767
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e5 \
6868
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
6969
--discriminator-size=64
7070
71+
# The newly trained model is output into this file...
72+
ls ne4x-custom-*.pkl.bz2
73+
7174
7275
.. image:: docs/BankLobby_example.gif
7376

0 commit comments

Comments
 (0)