diff options
| author | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-04 16:07:04 +0200 |
|---|---|---|
| committer | Marcin Chrzanowski <m@m-chrzan.xyz> | 2021-05-04 16:07:04 +0200 |
| commit | 1dd6898f248ab4085e89c3bfd14c8fd6b31d2839 (patch) | |
| tree | 29a411b802781f003b31d8bc928feaff1f2044a4 | |
| parent | c3297da401dd13289adb1e2bc713976d6108294a (diff) | |
Add augmentation experiment configs
| -rw-r--r-- | experiments/augment/0.yaml | 38 | ||||
| -rw-r--r-- | experiments/augment/1.yaml | 40 | ||||
| -rw-r--r-- | experiments/augment/3.yaml | 45 |
3 files changed, 123 insertions, 0 deletions
diff --git a/experiments/augment/0.yaml b/experiments/augment/0.yaml new file mode 100644 index 0000000..68e171f --- /dev/null +++ b/experiments/augment/0.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 diff --git a/experiments/augment/1.yaml b/experiments/augment/1.yaml new file mode 100644 index 0000000..42fb753 --- /dev/null +++ b/experiments/augment/1.yaml @@ -0,0 +1,40 @@ +type: classification +lr: 0.01 +batch_size: 100 +batch_norm: true +epochs: 30 +augmentations: + - rotation: 0 + - rotation: 1 + vflip: true +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 diff --git a/experiments/augment/3.yaml b/experiments/augment/3.yaml new file mode 100644 index 0000000..86c64fe --- /dev/null +++ b/experiments/augment/3.yaml @@ -0,0 +1,45 @@ +type: classification +lr: 0.01 +batch_size: 100 +batch_norm: true +epochs: 30 +augmentations: + - rotation: 0 + - rotation: 1 + vflip: true + - rotation: 1 + vflip: true + hflip: true + - rotation: 2 + hflip: true +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 |