Hugging Face's logo

Spaces:
waidhoferj
/
dance-classifier
Runtime error

App Files Community
waidhoferj commited on
Commit
c95adc4
·
1 Parent(s): 797a86a

updated remote training yaml

Browse files
Files changed (1) hide show
  1. models/config/train.yaml +1 -2
models/config/train.yaml CHANGED
@@ -18,7 +18,7 @@ dance_ids: &dance_ids
18
 
19
  data_module:
20
  batch_size: 64
21
- num_workers: 10
22
  test_proportion: 0.2
23
 
24
  datasets:
@@ -50,4 +50,3 @@ trainer:
50
 
51
  training_environment:
52
  learning_rate: 0.00053
53
- log_spectrograms: False
 
18
 
19
  data_module:
20
  batch_size: 64
21
+ num_workers: 5
22
  test_proportion: 0.2
23
 
24
  datasets:
 
50
 
51
  training_environment:
52
  learning_rate: 0.00053