1 files changed, 5 insertions, 0 deletions

diff --git a/go.sh b/go.sh
new file mode 100644
index 0000000..1f2fc26
--- /dev/null
+++ b/go.sh
@@ -0,0 +1,5 @@
+#!/bin/sh
+
+# ./training/train_cnn.py --layers 3 --kernel_sizes 3,3,3 --epochs 10000 --batch_size 16 --input training/input/ --target training/target_2/ --checkpoint-every 1000
+./training/train_cnn.py --export-only training/checkpoints/checkpoint_epoch_2000.pth
+./training/train_cnn.py --export-only training/checkpoints/checkpoint_epoch_2000.pth --infer training/input/img_001.png --output test/toto.png