Hugging Face's logo

Spaces:
waidhoferj
/
dance-classifier
Runtime error

App Files Community
dance-classifier
File size: 831 Bytes 
557fb53
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
global:
  id: decision_tree
  device: mps
  seed: 42
  dance_ids:
    - ATN
    - BCH
    - CHA
    - ECS
    - HST
    - JIV
    - QST
    - RMB
    - SFT
    - SLS
    - SMB
    - SWZ
    - TGO
    - VWZ
    - WCS
data_module:
  song_data_path: data/songs_cleaned.csv
  song_audio_path: data/samples
  batch_size: 32
  num_workers: 7
  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

trainer:
  log_every_n_steps: 15
  accelerator: gpu
  max_epochs: 50
  min_epochs: 5
  fast_dev_run: False
  # gradient_clip_val: 0.5
  # overfit_batches: 1
training_environment:
  learning_rate: 0.00053
model:
  n_channels: 128