artyomboyko commited on
Commit
7fa431c
1 Parent(s): 571b0a1

Update README.md

Browse files

Loaded the best model manually because Trainer chose a worse model. The parameter load_best_model_at_end=True did not work.

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
18
 
19
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.5759
22
- - Accuracy: 0.87
23
 
24
  ## Model description
25
 
 
18
 
19
  This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the GTZAN dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.4989
22
+ - Accuracy: 0.91
23
 
24
  ## Model description
25