Add training scripts for networks currently being trained, for release v0.2.

main
Alex J. Champandard 9 years ago
parent 1ad40b6d71
commit cabaaeeefe

@ -0,0 +1,18 @@
#!/bin/sh
python3.4 enhance.py \
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
--epochs=50 --batch-shape=192 --device=gpu0 \
--generator-downscale=2 --generator-upscale=2 \
--generator-blocks=8 --generator-filters=64 \
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
--train-blur=3 --train-noise=5.0
python3.4 enhance.py \
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
--epochs=500 --batch-shape=192 --device=gpu0 \
--generator-downscale=2 --generator-upscale=2 \
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
--discriminator-size=32 \
--train-blur=3 --train-noise=5.0

@ -0,0 +1,18 @@
#!/bin/sh
python3.4 enhance.py \
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
--epochs=50 --batch-shape=192 --device=gpu0 \
--generator-downscale=1 --generator-upscale=2 \
--generator-blocks=8 --generator-filters=64 \
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
--train-blur=2 --train-noise=4.0
python3.4 enhance.py \
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
--epochs=500 --batch-shape=192 --device=gpu0 \
--generator-downscale=1 --generator-upscale=2 \
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
--discriminator-size=32 \
--train-blur=2 --train-noise=4.0

@ -0,0 +1,18 @@
#!/bin/sh
python3.4 enhance.py \
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
--epochs=50 --batch-shape=192 --device=gpu0 \
--generator-downscale=0 --generator-upscale=2 \
--generator-blocks=8 --generator-filters=64 \
--perceptual-layer=conv2_2 --smoothness-weight=1e7 --adversary-weight=0.0 \
--train-blur=1 --train-noise=3.0
python3.4 enhance.py \
--train "$OPEN_IMAGES_PATH/*/*.jpg" --model small \
--epochs=500 --batch-shape=192 --device=gpu0 \
--generator-downscale=0 --generator-upscale=2 \
--perceptual-layer=conv5_2 --smoothness-weight=2e4 --adversary-weight=2e2 \
--generator-start=5 --discriminator-start=0 --adversarial-start=5 \
--discriminator-size=32 \
--train-blur=1 --train-noise=3.0
Loading…
Cancel
Save