m-chrzan.xyz
aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--experiments/depth/deep.yaml36
-rw-r--r--experiments/depth/medium.yaml30
-rw-r--r--experiments/depth/shallow.yaml22
3 files changed, 88 insertions, 0 deletions
diff --git a/experiments/depth/deep.yaml b/experiments/depth/deep.yaml
new file mode 100644
index 0000000..89e902c
--- /dev/null
+++ b/experiments/depth/deep.yaml
@@ -0,0 +1,36 @@
+type: counting-big
+lr: 0.01
+batch_size: 100
+batch_norm: true
+epochs: 16
+augmentations:
+ - rotation: 0
+convolutions:
+ - in_channels: 1
+ out_channels: 8
+ max_pool: true
+ - in_channels: 8
+ out_channels: 16
+ - in_channels: 16
+ out_channels: 32
+ max_pool: true
+ - in_channels: 32
+ out_channels: 64
+ - in_channels: 64
+ out_channels: 128
+ - in_channels: 128
+ out_channels: 256
+ max_pool: true
+ - in_channels: 256
+ out_channels: 512
+linears:
+ - in_features: 4608
+ out_features: 1024
+ - in_features: 1024
+ out_features: 512
+ - in_features: 512
+ out_features: 256
+ - in_features: 256
+ out_features: 120
+ - in_features: 120
+ out_features: 84
diff --git a/experiments/depth/medium.yaml b/experiments/depth/medium.yaml
new file mode 100644
index 0000000..2daf590
--- /dev/null
+++ b/experiments/depth/medium.yaml
@@ -0,0 +1,30 @@
+type: counting-big
+lr: 0.01
+batch_size: 100
+batch_norm: true
+epochs: 16
+augmentations:
+ - rotation: 0
+convolutions:
+ - in_channels: 1
+ out_channels: 8
+ max_pool: true
+ - in_channels: 8
+ out_channels: 16
+ - in_channels: 16
+ out_channels: 32
+ max_pool: true
+ - in_channels: 32
+ out_channels: 64
+ max_pool: true
+ - in_channels: 64
+ out_channels: 128
+linears:
+ - in_features: 1152
+ out_features: 512
+ - in_features: 512
+ out_features: 256
+ - in_features: 256
+ out_features: 120
+ - in_features: 120
+ out_features: 84
diff --git a/experiments/depth/shallow.yaml b/experiments/depth/shallow.yaml
new file mode 100644
index 0000000..8ae8c90
--- /dev/null
+++ b/experiments/depth/shallow.yaml
@@ -0,0 +1,22 @@
+type: counting-big
+lr: 0.01
+batch_size: 100
+batch_norm: true
+epochs: 16
+augmentations:
+ - rotation: 0
+convolutions:
+ - in_channels: 1
+ out_channels: 8
+ max_pool: true
+ - in_channels: 8
+ out_channels: 16
+ max_pool: true
+ - in_channels: 16
+ out_channels: 32
+ max_pool: true
+linears:
+ - in_features: 288
+ out_features: 120
+ - in_features: 120
+ out_features: 84