From faf5ff5c98b55a55f697f73c0da26c2643fd976a Mon Sep 17 00:00:00 2001
From: Pablo Olivares <pablolivares1502@gmail.com>
Date: Tue, 19 Mar 2024 00:22:15 +0100
Subject: [PATCH] Config file created advances #2

This config file will make the project easier to maintain and the models easier to train.
---
 config/config.yaml | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 config/config.yaml

diff --git a/config/config.yaml b/config/config.yaml
new file mode 100644
index 0000000..179e0cd
--- /dev/null
+++ b/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/"