update cloud run config

models/config/train.yaml

@@ -1,47 +1,53 @@
-global:
-  id: ast_ptl
-  device: cuda
-  seed: 42
-  dance_ids:
-    - ATN
-    - BCH
-    - CHA
-    - ECS
-    - HST
-    - JIV
-    - QST
-    - RMB
-    - SFT
-    - SLS
-    - SMB
-    - SWZ
-    - TGO
-    - VWZ
-    - WCS
+training_fn: audio_spectrogram_transformer.train_lightning_ast
+device: cuda
+seed: 42
+dance_ids: &dance_ids
+  - BCH
+  - CHA
+  - JIV
+  - ECS
+  - QST
+  - RMB
+  - SFT
+  - SLS
+  - SMB
+  - SWZ
+  - TGO
+  - VWZ
+  - WCS
+
 data_module:
-  song_data_path: ../datastores/dance-music/songs_cleaned.csv
-  song_audio_path: ../datastores/dance-music
   batch_size: 64
-  num_workers: 4
-  min_votes: 1
-  dataset_kwargs:
-    audio_window_duration: 6
-    audio_window_jitter: 1.5
-    # audio_pipeline_kwargs:
-    #   mask_count: 0 # Don't mask the data
-    #   snr_mean: 15.0 # Pretty much eliminate the noise
-    #   freq_mask_size: 10
-    #   time_mask_size: 80
+  num_workers: 10
+  test_proportion: 0.2
+
+datasets:
+  preprocessing.dataset.Music4DanceDataset:
+    song_data_path: ../datastores/dance-music/songs_cleaned.csv
+    song_audio_path: ../datastores/dance-music
+    class_list: *dance_ids
+    multi_label: False
+    min_votes: 1
+    audio_window_jitter: 0.7
+
+model:
+  n_channels: 128
+
+feature_extractor:
+  mask_count: 0 # Don't mask the data
+  snr_mean: 15.0 # Pretty much eliminate the noise
+  freq_mask_size: 10
+  time_mask_size: 80
 
 trainer:
   log_every_n_steps: 15
   accelerator: gpu
   max_epochs: 50
-  min_epochs: 5
+  min_epochs: 7
   fast_dev_run: False
   # gradient_clip_val: 0.5
-  overfit_batches: 1
+  # overfit_batches: 1
+
 training_environment:
   learning_rate: 0.00053
-model:
-  n_channels: 128
+  log_spectrograms: False