Hugging Face's logo

Spaces:
waidhoferj
/
dance-classifier
Runtime error

App Files Community
waidhoferj commited on
Commit
4427a9f
·
1 Parent(s): d78d6d1

prepared for wav2vec2 training

Browse files
Files changed (1) hide show
  1. models/config/train.yaml +1 -1
models/config/train.yaml CHANGED
@@ -1,4 +1,4 @@
1
- training_fn: audio_spectrogram_transformer.train_lightning_ast
2
  device: cuda
3
  seed: 42
4
  dance_ids: &dance_ids
 
1
+ training_fn: wav2vec2.train_huggingface
2
  device: cuda
3
  seed: 42
4
  dance_ids: &dance_ids