abhishtagatya commited on
Commit
e75371b
1 Parent(s): 0d56164

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -5
README.md CHANGED
@@ -22,13 +22,24 @@ This model is a fine-tuned version of [facebook/wav2vec2-base-960h](https://hugg
22
  It achieves the following results on the evaluation set:
23
  - Loss: 0.0917
24
  - Accuracy: 0.9835
25
- - Far: 0.0068
26
- - Frr: 0.0330
27
- - Eer: 0.0199
28
 
29
  ## Model description
30
 
31
- More information needed
 
 
 
 
 
 
 
 
 
 
 
32
 
33
  ## Intended uses & limitations
34
 
@@ -55,7 +66,7 @@ The following hyperparameters were used during training:
55
 
56
  ### Training results
57
 
58
- | Training Loss | Epoch | Step | Validation Loss | Accuracy | Far | Frr | Eer |
59
  |:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:------:|:------:|
60
  | 0.6363 | 0.39 | 2500 | 0.4678 | 0.8652 | 0.0178 | 0.3326 | 0.1752 |
61
  | 0.2896 | 0.79 | 5000 | 0.1145 | 0.9744 | 0.0170 | 0.0402 | 0.0286 |
 
22
  It achieves the following results on the evaluation set:
23
  - Loss: 0.0917
24
  - Accuracy: 0.9835
25
+ - FAR: 0.0068
26
+ - FRR: 0.0330
27
+ - EER: 0.0199
28
 
29
  ## Model description
30
 
31
+ ### Quick Use
32
+
33
+ ```python
34
+ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
35
+
36
+ config = AutoConfig.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
37
+ feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake")
38
+
39
+ model = HubertForSequenceClassification.from_pretrained("abhishtagatya/hubert-base-960h-itw-deepfake", config=config).to(device)
40
+
41
+ # Your Logic Here
42
+ ```
43
 
44
  ## Intended uses & limitations
45
 
 
66
 
67
  ### Training results
68
 
69
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | FAR | FRR | EER |
70
  |:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:------:|:------:|
71
  | 0.6363 | 0.39 | 2500 | 0.4678 | 0.8652 | 0.0178 | 0.3326 | 0.1752 |
72
  | 0.2896 | 0.79 | 5000 | 0.1145 | 0.9744 | 0.0170 | 0.0402 | 0.0286 |