Add base net config

1 files changed, 38 insertions, 0 deletions

diff --git a/experiments/net.yaml b/experiments/net.yaml
new file mode 100644
index 0000000..68e171f
--- /dev/null
+++ b/experiments/net.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