soumya13 commited on
Commit
85dab20
·
1 Parent(s): e7bd887

Training in progress epoch 0

Browse files
Files changed (3) hide show
  1. README.md +5 -29
  2. config.json +24 -24
  3. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,10 +14,10 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 0.0230
18
- - Validation Loss: 0.0002
19
- - Train Accuracy: 1.0
20
- - Epoch: 24
21
 
22
  ## Model description
23
 
@@ -43,31 +43,7 @@ The following hyperparameters were used during training:
43
 
44
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
45
  |:----------:|:---------------:|:--------------:|:-----:|
46
- | 2.5515 | 1.0912 | 0.5385 | 0 |
47
- | 0.6884 | 0.4192 | 0.9231 | 1 |
48
- | 0.3177 | 0.1240 | 1.0 | 2 |
49
- | 0.1737 | 0.0375 | 1.0 | 3 |
50
- | 0.0903 | 0.0094 | 1.0 | 4 |
51
- | 0.0804 | 0.0047 | 1.0 | 5 |
52
- | 0.0578 | 0.0029 | 1.0 | 6 |
53
- | 0.0234 | 0.0022 | 1.0 | 7 |
54
- | 0.0157 | 0.0015 | 1.0 | 8 |
55
- | 0.0301 | 0.0012 | 1.0 | 9 |
56
- | 0.0261 | 0.0009 | 1.0 | 10 |
57
- | 0.0244 | 0.0008 | 1.0 | 11 |
58
- | 0.0362 | 0.0007 | 1.0 | 12 |
59
- | 0.0133 | 0.0006 | 1.0 | 13 |
60
- | 0.0185 | 0.0005 | 1.0 | 14 |
61
- | 0.0256 | 0.0004 | 1.0 | 15 |
62
- | 0.0224 | 0.0004 | 1.0 | 16 |
63
- | 0.0111 | 0.0003 | 1.0 | 17 |
64
- | 0.0057 | 0.0003 | 1.0 | 18 |
65
- | 0.0136 | 0.0003 | 1.0 | 19 |
66
- | 0.0167 | 0.0002 | 1.0 | 20 |
67
- | 0.0172 | 0.0002 | 1.0 | 21 |
68
- | 0.0066 | 0.0002 | 1.0 | 22 |
69
- | 0.0158 | 0.0002 | 1.0 | 23 |
70
- | 0.0230 | 0.0002 | 1.0 | 24 |
71
 
72
 
73
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [gpt2](https://huggingface.co/gpt2) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 4.0208
18
+ - Validation Loss: 1.7346
19
+ - Train Accuracy: 0.3590
20
+ - Epoch: 0
21
 
22
  ## Model description
23
 
 
43
 
44
  | Train Loss | Validation Loss | Train Accuracy | Epoch |
45
  |:----------:|:---------------:|:--------------:|:-----:|
46
+ | 4.0208 | 1.7346 | 0.3590 | 0 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
47
 
48
 
49
  ### Framework versions
config.json CHANGED
@@ -9,33 +9,33 @@
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 50256,
11
  "id2label": {
12
- "0": "autonomous",
13
- "1": "hyundai",
14
- "2": "nissan",
15
- "3": "mercedes-benz",
16
- "4": "lexus",
17
- "5": "chrysler",
18
- "6": "cruise",
19
- "7": "ford",
20
- "8": "jaguar",
21
- "9": "chevrolet",
22
- "10": "toyota",
23
- "11": "lincoln"
24
  },
25
  "initializer_range": 0.02,
26
  "label2id": {
27
- "autonomous": 0,
28
- "chevrolet": 9,
29
- "chrysler": 5,
30
- "cruise": 6,
31
- "ford": 7,
32
- "hyundai": 1,
33
- "jaguar": 8,
34
- "lexus": 4,
35
- "lincoln": 11,
36
- "mercedes-benz": 3,
37
- "nissan": 2,
38
- "toyota": 10
39
  },
40
  "layer_norm_epsilon": 1e-05,
41
  "model_type": "gpt2",
 
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 50256,
11
  "id2label": {
12
+ "0": "mercedes-benz",
13
+ "1": "cruise",
14
+ "2": "toyota",
15
+ "3": "chrysler",
16
+ "4": "ford",
17
+ "5": "hyundai",
18
+ "6": "autonomous",
19
+ "7": "nissan",
20
+ "8": "chevrolet",
21
+ "9": "lincoln",
22
+ "10": "jaguar",
23
+ "11": "lexus"
24
  },
25
  "initializer_range": 0.02,
26
  "label2id": {
27
+ "autonomous": 6,
28
+ "chevrolet": 8,
29
+ "chrysler": 3,
30
+ "cruise": 1,
31
+ "ford": 4,
32
+ "hyundai": 5,
33
+ "jaguar": 10,
34
+ "lexus": 11,
35
+ "lincoln": 9,
36
+ "mercedes-benz": 0,
37
+ "nissan": 7,
38
+ "toyota": 2
39
  },
40
  "layer_norm_epsilon": 1e-05,
41
  "model_type": "gpt2",
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6bac8500dc9b1fa24fa6d7f7259d4833c01806ff8b0f31c9bd8d98ba272d5687
3
  size 497983984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b6412f7a4192e26441297299fe2af9e222fbe00e32998a0c407266fa32634d6
3
  size 497983984