From 8aa3463c48ccc43f98ecb3e67f18ca8f139fe86c Mon Sep 17 00:00:00 2001 From: Marcin Chrzanowski Date: Mon, 3 May 2021 22:30:38 +0200 Subject: Add depth experiment configs --- experiments/depth/deep.yaml | 36 ++++++++++++++++++++++++++++++++++++ experiments/depth/medium.yaml | 30 ++++++++++++++++++++++++++++++ experiments/depth/shallow.yaml | 22 ++++++++++++++++++++++ 3 files changed, 88 insertions(+) create mode 100644 experiments/depth/deep.yaml create mode 100644 experiments/depth/medium.yaml create mode 100644 experiments/depth/shallow.yaml 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 -- cgit v1.2.3