From 2511b220955998e088964b7ce01b5714bea50e29 Mon Sep 17 00:00:00 2001 From: Marcin Chrzanowski Date: Tue, 4 May 2021 16:08:14 +0200 Subject: Add base net config --- experiments/net.yaml | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 experiments/net.yaml 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 -- cgit v1.2.3