Add training configs to version control

glinscott · May 16, 2018 · 46ecdc7 · 46ecdc7
1 parent 4f575e0
commit 46ecdc7
Show file tree

Hide file tree

Showing 3 changed files with 81 additions and 0 deletions.
diff --git a/training/tf/configs/279-286.yaml b/training/tf/configs/279-286.yaml
@@ -0,0 +1,27 @@
+%YAML 1.2
+---
+name: 'online-192x15'
+gpu: 0
+
+dataset:
+    num_chunks: 500000
+    train_ratio: 0.90
+    input: '/home/folkert/data/run1/'
+
+training:
+    batch_size: 1024
+    total_steps: 50000
+    shuffle_size: 1000000
+    lr_values:
+        - 0.0005
+        - 0.0001
+    lr_boundaries:
+        - 25000
+    policy_loss_weight: 1.0
+    value_loss_weight: 1.0
+    path: '/home2/networks'
+
+model:
+    filters: 192
+    residual_blocks: 15
+...
diff --git a/training/tf/configs/287-301.yaml b/training/tf/configs/287-301.yaml
@@ -0,0 +1,27 @@
+%YAML 1.2
+---
+name: 'online-192x15'
+gpu: 0
+
+dataset:
+    num_chunks: 500000
+    train_ratio: 0.90
+    input: '/home/folkert/data/run1/'
+
+training:
+    batch_size: 1024
+    total_steps: 50000
+    shuffle_size: 1000000
+    lr_values:
+        - 0.0005
+        - 0.0001
+    lr_boundaries:
+        - 25000
+    policy_loss_weight: 1.0
+    value_loss_weight: 0.25
+    path: '/home2/networks'
+
+model:
+    filters: 192
+    residual_blocks: 15
+...
diff --git a/training/tf/configs/current.yaml b/training/tf/configs/current.yaml
@@ -0,0 +1,27 @@
+%YAML 1.2
+---
+name: 'online-192x15'
+gpu: 0
+
+dataset:
+    num_chunks: 500000
+    train_ratio: 0.90
+    input: '/home/folkert/data/run1/'
+
+training:
+    batch_size: 1024
+    total_steps: 50000
+    shuffle_size: 1000000
+    lr_values:
+        - 0.0005
+        - 0.0001
+    lr_boundaries:
+        - 10000
+    policy_loss_weight: 1.0 
+    value_loss_weight: 0.25
+    path: '/home2/networks'
+
+model:
+    filters: 192 
+    residual_blocks: 15
+...