End of training

README.md

@@ -1,4 +1,5 @@
 ---
+library_name: transformers
 license: apache-2.0
 base_model: ntu-spml/distilhubert
 tags:
@@ -32,7 +33,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [ntu-spml/distilhubert](https://huggingface.co/ntu-spml/distilhubert) on the audiofolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6680
+- Loss: 0.6682
 - Accuracy: 0.8370
 
 ## Model description
@@ -66,13 +67,13 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| No log        | 0.8696 | 5    | 0.6810          | 0.8370   |
-| No log        | 1.7391 | 10   | 0.6680          | 0.8370   |
+| No log        | 0.8696 | 5    | 0.6812          | 0.8370   |
+| No log        | 1.7391 | 10   | 0.6682          | 0.8370   |
 
 
 ### Framework versions
 
-- Transformers 4.44.0
-- Pytorch 2.4.0+cpu
+- Transformers 4.44.2
+- Pytorch 2.4.0+cu118
 - Datasets 2.21.0
 - Tokenizers 0.19.1

config.json

@@ -80,7 +80,7 @@
   "num_hidden_layers": 2,
   "pad_token_id": 0,
   "torch_dtype": "float32",
-  "transformers_version": "4.44.0",
+  "transformers_version": "4.44.2",
   "use_weighted_layer_sum": false,
   "vocab_size": 32
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:918e229fcc739079e678d896ed5a3a445cf43b7d2dade6505293786b8a9883be
+oid sha256:518cfbb7e0b0273a041bf9f389c2e399583fd8d6dc466f5a9a47f94dac442b10
 size 94766588

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0167f1fac8e0ba987a495bed49c28f68834844f12647759f1a9b592c9e7986a3
+oid sha256:497d1576d4deff8c5fe322b69928b2b277e2f4e35af91aa87223620fc912bf0c
 size 5176