From 742cd10f8b96718d711c785ab9472bc5b7a92c2a Mon Sep 17 00:00:00 2001
From: Marcin Chrzanowski <m@m-chrzan.xyz>
Date: Tue, 4 May 2021 16:07:43 +0200
Subject: Add batch norm experiment configs

---
 experiments/batch_norm/no.yaml  | 38 ++++++++++++++++++++++++++++++++++++++
 experiments/batch_norm/yes.yaml | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 experiments/batch_norm/no.yaml
 create mode 100644 experiments/batch_norm/yes.yaml

(limited to 'experiments')

diff --git a/experiments/batch_norm/no.yaml b/experiments/batch_norm/no.yaml
new file mode 100644
index 0000000..7d6ff42
--- /dev/null
+++ b/experiments/batch_norm/no.yaml
@@ -0,0 +1,38 @@
+type: classification
+lr: 0.01
+batch_size: 100
+batch_norm: false
+epochs: 30
+augmentations:
+  - rotation: 0
+convolutions:
+  - in_channels: 1
+    out_channels: 8
+    size: 11
+    padding: 5
+  - in_channels: 8
+    out_channels: 16
+  - in_channels: 16
+    out_channels: 16
+  - in_channels: 16
+    out_channels: 16
+    max_pool: true
+  - in_channels: 16
+    out_channels: 32
+    max_pool: true
+  - in_channels: 32
+    out_channels: 64
+    max_pool: true
+  - in_channels: 64
+    out_channels: 128
+  - in_channels: 128
+    out_channels: 256
+linears:
+  - in_features: 2304
+    out_features: 1024
+  - in_features: 1024
+    out_features: 512
+  - in_features: 512
+    out_features: 150
+  - in_features: 150
+    out_features: 150
diff --git a/experiments/batch_norm/yes.yaml b/experiments/batch_norm/yes.yaml
new file mode 100644
index 0000000..68e171f
--- /dev/null
+++ b/experiments/batch_norm/yes.yaml
@@ -0,0 +1,38 @@
+type: classification
+lr: 0.01
+batch_size: 100
+batch_norm: true
+epochs: 30
+augmentations:
+  - rotation: 0
+convolutions:
+  - in_channels: 1
+    out_channels: 8
+    size: 11
+    padding: 5
+  - in_channels: 8
+    out_channels: 16
+  - in_channels: 16
+    out_channels: 16
+  - in_channels: 16
+    out_channels: 16
+    max_pool: true
+  - in_channels: 16
+    out_channels: 32
+    max_pool: true
+  - in_channels: 32
+    out_channels: 64
+    max_pool: true
+  - in_channels: 64
+    out_channels: 128
+  - in_channels: 128
+    out_channels: 256
+linears:
+  - in_features: 2304
+    out_features: 1024
+  - in_features: 1024
+    out_features: 512
+  - in_features: 512
+    out_features: 150
+  - in_features: 150
+    out_features: 150
-- 
cgit v1.2.3