m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
path: root/experiments
diff options
context:
space:
mode:
Diffstat (limited to 'experiments')
-rw-r--r--experiments/net.yaml38
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