add img configs

Author: slegroux

config/data/image/fashion_mnist.yaml

@@ -0,0 +1,15 @@
+_target_: nimrod.image.datasets.ImageDataModule
+name: "fashion_mnist"
+data_dir: "../data/image"
+train_val_split: [0.8, 0.2]
+batch_size: 128
+num_workers: 0
+pin_memory: False
+persistent_workers: False
+transforms: 
+  _target_: torchvision.transforms.Compose
+  transforms:
+    - _target_: torchvision.transforms.ToTensor
+    - _target_: torchvision.transforms.Resize
+      size: 32
+

config/data/image/smithsonian_butterflies.yaml

@@ -0,0 +1,17 @@
+_target_: nimrod.image.datasets.ImageDataModule
+name: "huggan/smithsonian_butterflies_subset"
+data_dir: "../data/image"
+train_val_split: [0.8, 0.2]
+batch_size: 128
+num_workers: 0
+pin_memory: False
+persistent_workers: False
+transforms: 
+  _target_: torchvision.transforms.Compose
+  transforms:
+    - _target_: torchvision.transforms.ToTensor
+    - _target_: torchvision.transforms.Resize
+      size: [32,32]
+    - _target_: torchvision.transforms.Normalize
+      mean: [0.5]
+      std: [0.5]

config/model/image/sequentialconv.yaml

@@ -0,0 +1,27 @@
+defaults:
+  - _self_
+
+model:
+  - _target_: nimrod.models.conv.ConvLayer
+    in_channels: 1
+    out_channels: 16
+    kernel_size: 3
+    bias: true
+    normalization:
+      _target_: hydra.utils.get_class
+      path: torch.nn.BatchNorm2d
+    activation:
+      _target_: hydra.utils.get_class
+      path: torch.nn.ReLU
+
+  - _target_: nimrod.models.conv.ConvLayer
+    in_channels: 16
+    out_channels: 32
+    kernel_size: 3
+    bias: true
+    normalization:
+      _target_: hydra.utils.get_class
+      path: torch.nn.BatchNorm2d
+    activation:
+      _target_: hydra.utils.get_class
+      path: torch.nn.ReLU