Update yahoo dataset config

zamanzadeh · Aug 27, 2024 · 5b89336 · 5b89336
1 parent 394d484
commit 5b89336
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/configs/pretext/carla_pretext_ucr.yml → configs/pretext/carla_pretext_yahoo.yml b/configs/pretext/carla_pretext_ucr.yml → configs/pretext/carla_pretext_yahoo.yml
@@ -8,9 +8,8 @@ model_kwargs:
    features_dim: 4
 
 # Dataset
-train_db_name: Power
-val_db_name: Power
-ds_name: '152_UCR_Anomaly_PowerDemand1_9000_18485_18821.txt'
+train_db_name: yahoo
+val_db_name: yahoo
 num_classes: 10
 
 # Loss
@@ -29,7 +28,7 @@ optimizer_kwargs:
 scheduler: cosine
 scheduler_kwargs:
    lr_decay_rate: 0.01
-batch_size: 100
+batch_size: 50
 num_workers: 1
 
 # Transformations