Spaces:
Runtime error
Runtime error
| training_fn: audio_spectrogram_transformer.train_lightning_ast | |
| device: cuda | |
| seed: 42 | |
| dance_ids: | |
| - BCH | |
| - CHA | |
| - JIV | |
| - ECS | |
| - QST | |
| - RMB | |
| - SFT | |
| - SLS | |
| - SMB | |
| - SWZ | |
| - TGO | |
| - VWZ | |
| - WCS | |
| data_module: | |
| batch_size: 64 | |
| 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: | |
| 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: 7 | |
| fast_dev_run: False | |
| # gradient_clip_val: 0.5 | |
| # overfit_batches: 1 | |
| training_environment: | |
| learning_rate: 0.00053 | |
| log_spectrograms: False | |