From dba6d2d91898d0a85422f825448972e30e036cb6 Mon Sep 17 00:00:00 2001
From: mbsantiago <santiago.mbal@gmail.com>
Date: Wed, 27 Aug 2025 23:44:49 +0100
Subject: [PATCH] Updating configs

---
 example_data/config.yaml   | 33 +++++++++++++++------------------
 example_data/dataset.yaml  |  8 ++++++++
 example_data/datasets.yaml |  0
 3 files changed, 23 insertions(+), 18 deletions(-)
 create mode 100644 example_data/dataset.yaml
 delete mode 100644 example_data/datasets.yaml

diff --git a/example_data/config.yaml b/example_data/config.yaml
index 1e12018..266746a 100644
--- a/example_data/config.yaml
+++ b/example_data/config.yaml
@@ -1,14 +1,3 @@
-datasets:
-  train:
-    name: example dataset
-    description: Only for demonstration purposes
-    sources:
-      - format: batdetect2
-        name: Example Data
-        description: Examples included for testing batdetect2
-        annotations_dir: example_data/anns
-        audio_dir: example_data/audio
-
 targets:
   classes:
     classes:
@@ -114,9 +103,19 @@ model:
             out_channels: 32
 
 train:
-  batch_size: 8
   learning_rate: 0.001
   t_max: 100
+
+  dataloaders:
+    train:
+      batch_size: 8
+      num_workers: 2
+      shuffle: True
+
+    val:
+      batch_size: 8
+      num_workers: 2
+
   loss:
     detection:
       weight: 1.0
@@ -130,14 +129,12 @@ train:
         alpha: 2
     size:
       weight: 0.1
+
   logger:
-    logger_type: mlflow
-    experiment_name: batdetect2
-    tracking_uri: http://localhost:5000
-    log_model: true
+    logger_type: csv
     save_dir: outputs/log/
-    artifact_location: outputs/artifacts/
-    checkpoint_path_prefix: outputs/checkpoints/
+    name: logs
+
   augmentations:
     steps:
       - augmentation_type: mix_audio
diff --git a/example_data/dataset.yaml b/example_data/dataset.yaml
new file mode 100644
index 0000000..790da8c
--- /dev/null
+++ b/example_data/dataset.yaml
@@ -0,0 +1,8 @@
+name: example dataset
+description: Only for demonstration purposes
+sources:
+  - format: batdetect2
+    name: Example Data
+    description: Examples included for testing batdetect2
+    annotations_dir: example_data/anns
+    audio_dir: example_data/audio
diff --git a/example_data/datasets.yaml b/example_data/datasets.yaml
deleted file mode 100644
index e69de29..0000000