Merge branch 'master' of github.com:awni/cinc17

awni · Mar 21, 2017 · 9b0ea20 · 9b0ea20
2 parents 0014e97 + 855e7ed
commit 9b0ea20
Show file tree

Hide file tree

Showing 2 changed files with 67 additions and 2 deletions.
diff --git a/configs/train.json b/configs/train.json
@@ -8,7 +8,7 @@
 
  "optimizer" : {
  "epochs" : 50,
- "learning_rate" : 1e-1,
+ "learning_rate" : 1e-3,
  "momentum" : 0.95,
  "decay_rate" : 1.0,
  "decay_steps" : 2000
@@ -18,6 +18,66 @@
  "dropout" : 0.0,
  "batch_size" : 32,
  "conv_layers" : [
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
+ { "kernel_size" : 64,
+ "num_filters" : 64,
+ "stride" : 2
+ },
  { "kernel_size" : 64,
  "num_filters" : 64,
  "stride" : 2
@@ -30,6 +90,6 @@
  },
 
  "io" : {
- "save_path" : "saved_models/model1"
+ "save_path" : "saved_models/cnn16-lr-3"
  }
 }
diff --git a/launch.sh b/launch.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+
+gpu=3
+env CUDA_VISIBLE_DEVICES=$gpu python train.py configs/cnn20-lr-1.json &
+echo "Started training at: ", $(date +%m:%d:%y-%H:%M:%S)