diff options
| author | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-03 22:30:38 +0200 |
|---|---|---|
| committer | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-03 22:30:42 +0200 |
| commit | 8aa3463c48ccc43f98ecb3e67f18ca8f139fe86c (patch) | |
| tree | 15a0cef4dbfbe7a463ce1f955edadb29c573d696 /experiments/depth/shallow.yaml | |
| parent | 3f45fb851bace1d5bb56cd4df06ba0f4c5c47b9b (diff) | |
Add depth experiment configs
Diffstat (limited to 'experiments/depth/shallow.yaml')
| -rw-r--r-- | experiments/depth/shallow.yaml | 22 |
1 files changed, 22 insertions, 0 deletions
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 |