From 1dd6898f248ab4085e89c3bfd14c8fd6b31d2839 Mon Sep 17 00:00:00 2001
From: Marcin Chrzanowski <m@m-chrzan.xyz>
Date: Tue, 4 May 2021 16:07:04 +0200
Subject: Add augmentation experiment configs

---
 experiments/augment/0.yaml | 38 ++++++++++++++++++++++++++++++++++++++
 experiments/augment/1.yaml | 40 ++++++++++++++++++++++++++++++++++++++++
 experiments/augment/3.yaml | 45 +++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 123 insertions(+)
 create mode 100644 experiments/augment/0.yaml
 create mode 100644 experiments/augment/1.yaml
 create mode 100644 experiments/augment/3.yaml

(limited to 'experiments')

diff --git a/experiments/augment/0.yaml b/experiments/augment/0.yaml
new file mode 100644
index 0000000..68e171f
--- /dev/null
+++ b/experiments/augment/0.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
diff --git a/experiments/augment/1.yaml b/experiments/augment/1.yaml
new file mode 100644
index 0000000..42fb753
--- /dev/null
+++ b/experiments/augment/1.yaml
@@ -0,0 +1,40 @@
+type: classification
+lr: 0.01
+batch_size: 100
+batch_norm: true
+epochs: 30
+augmentations:
+  - rotation: 0
+  - rotation: 1
+    vflip: true
+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/augment/3.yaml b/experiments/augment/3.yaml
new file mode 100644
index 0000000..86c64fe
--- /dev/null
+++ b/experiments/augment/3.yaml
@@ -0,0 +1,45 @@
+type: classification
+lr: 0.01
+batch_size: 100
+batch_norm: true
+epochs: 30
+augmentations:
+  - rotation: 0
+  - rotation: 1
+    vflip: true
+  - rotation: 1
+    vflip: true
+    hflip: true
+  - rotation: 2
+    hflip: true
+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