Bug fixes in testing

BojarLab · Aug 21, 2024 · 02e04d7 · 02e04d7
1 parent b8fa6aa
commit 02e04d7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/tests/dummy_data/datasets.yaml b/tests/dummy_data/datasets.yaml
@@ -1,6 +1,7 @@
 seed: 42
 data_dir: ./
 root_dir: ./data
+logs_dir: ./logs
 datasets:
   - name: class-1
     label: class
@@ -45,6 +46,7 @@ model:
   - name: sweetnet
     hidden_dim: 128
     batch_size: 256
+    num_layers: 3
     epochs: 5
     learning_rate: 0.001
     optimizer: Adam

diff --git a/tests/dummy_data/pre_transform.yaml b/tests/dummy_data/pre_transform.yaml
@@ -1,6 +1,7 @@
 seed: 42
 data_dir: ./
 root_dir: ./data
+logs_dir: ./logs
 datasets:
   - name: class-1
     label: class