Config file created advances #2

This config file will make the project easier to maintain and the models easier to train.

config/config.yaml

@@ -0,0 +1,17 @@
+model:
+  name: efficientnet_b0
+  num_classes: 10
+  pretrained: true
+
+training:
+  batch_size: 32
+  num_epochs: 2
+  learning_rate: 0.001
+
+data:
+  name: "CIFAR10"
+  dataset_path: "./data/"
+
+paths:
+  log_path: "./logs/"
+  plot_path: "./outputs/figures/"