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

Commit cabaaee

Browse files
committed
Add training scripts for networks currently being trained, for release v0.2.
1 parent 1ad40b6 commit cabaaee

File tree

3 files changed

+54
-0
lines changed

3 files changed

+54
-0
lines changed

scripts/small-1x.sh

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/sh
2+
3+
python3.4 enhance.py \
4+
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
5+
--epochs=50 --batch-shape=192 --device=gpu0 \
6+
--generator-downscale=2 --generator-upscale=2 \
7+
--generator-blocks=8 --generator-filters=64 \
8+
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
9+
--train-blur=3 --train-noise=5.0
10+
11+
python3.4 enhance.py \
12+
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
13+
--epochs=500 --batch-shape=192 --device=gpu0 \
14+
--generator-downscale=2 --generator-upscale=2 \
15+
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \
16+
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
17+
--discriminator-size=32 \
18+
--train-blur=3 --train-noise=5.0

scripts/small-2x.sh

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/sh
2+
3+
python3.4 enhance.py \
4+
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
5+
--epochs=50 --batch-shape=192 --device=gpu0 \
6+
--generator-downscale=1 --generator-upscale=2 \
7+
--generator-blocks=8 --generator-filters=64 \
8+
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
9+
--train-blur=2 --train-noise=4.0
10+
11+
python3.4 enhance.py \
12+
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
13+
--epochs=500 --batch-shape=192 --device=gpu0 \
14+
--generator-downscale=1 --generator-upscale=2 \
15+
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \
16+
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
17+
--discriminator-size=32 \
18+
--train-blur=2 --train-noise=4.0

scripts/small-4x.sh

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
#!/bin/sh
2+
3+
python3.4 enhance.py \
4+
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
5+
--epochs=50 --batch-shape=192 --device=gpu0 \
6+
--generator-downscale=0 --generator-upscale=2 \
7+
--generator-blocks=8 --generator-filters=64 \
8+
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
9+
--train-blur=1 --train-noise=3.0
10+
11+
python3.4 enhance.py \
12+
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
13+
--epochs=500 --batch-shape=192 --device=gpu0 \
14+
--generator-downscale=0 --generator-upscale=2 \
15+
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \
16+
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
17+
--discriminator-size=32 \
18+
--train-blur=1 --train-noise=3.0

0 commit comments

Comments
 (0)