jordyvl commited on
Commit
4849668
·
1 Parent(s): f1b5c99

Saving best model to hub

Browse files
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
all_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 99.96,
3
+ "train_loss": 4.706060151404296e+25,
4
+ "train_runtime": 15225.9839,
5
+ "train_samples_per_second": 5.254,
6
+ "train_steps_per_second": 0.039
7
+ }
config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/swinv2-base-patch4-window12-192-22k",
3
+ "architectures": [
4
+ "Swinv2ForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 18,
11
+ 2
12
+ ],
13
+ "drop_path_rate": 0.1,
14
+ "embed_dim": 128,
15
+ "encoder_stride": 32,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 1024,
19
+ "id2label": {
20
+ "0": "ADVE",
21
+ "1": "Email",
22
+ "2": "Form",
23
+ "3": "Letter",
24
+ "4": "Memo",
25
+ "5": "News",
26
+ "6": "Note",
27
+ "7": "Report",
28
+ "8": "Resume",
29
+ "9": "Scientific"
30
+ },
31
+ "image_size": 192,
32
+ "initializer_range": 0.02,
33
+ "label2id": {
34
+ "ADVE": 0,
35
+ "Email": 1,
36
+ "Form": 2,
37
+ "Letter": 3,
38
+ "Memo": 4,
39
+ "News": 5,
40
+ "Note": 6,
41
+ "Report": 7,
42
+ "Resume": 8,
43
+ "Scientific": 9
44
+ },
45
+ "layer_norm_eps": 1e-05,
46
+ "mlp_ratio": 4.0,
47
+ "model_type": "swinv2",
48
+ "num_channels": 3,
49
+ "num_heads": [
50
+ 4,
51
+ 8,
52
+ 16,
53
+ 32
54
+ ],
55
+ "num_layers": 4,
56
+ "patch_size": 4,
57
+ "path_norm": true,
58
+ "pretrained_window_sizes": [
59
+ 0,
60
+ 0,
61
+ 0,
62
+ 0
63
+ ],
64
+ "problem_type": "single_label_classification",
65
+ "qkv_bias": true,
66
+ "torch_dtype": "float32",
67
+ "transformers_version": "4.26.1",
68
+ "use_absolute_embeddings": false,
69
+ "window_size": 12
70
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:61470b105d5a4d2be762d55ee3a2410b73b02444254212c05a7367d3ac3e06fa
3
+ size 347784881
train_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 99.96,
3
+ "train_loss": 4.706060151404296e+25,
4
+ "train_runtime": 15225.9839,
5
+ "train_samples_per_second": 5.254,
6
+ "train_steps_per_second": 0.039
7
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d94697f2bc0a8d49aa8af756d7345c65214d38c20f0924885deb30e6f7d4996
3
+ size 3579