Chenghao-Qiu commited on
Commit
04bcd99
·
verified ·
1 Parent(s): 43e1d28

Upload folder using huggingface_hub

Browse files
config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "pGenomeOcean/GenomeOcean-100M",
3
+ "architectures": [
4
+ "MistralForSequenceClassification"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "auto_map": {
8
+ "AutoConfig": "pGenomeOcean/GenomeOcean-100M--configuration_mistral.MistralConfig",
9
+ "AutoModel": "pGenomeOcean/GenomeOcean-100M--modeling_mistral.MistralModel",
10
+ "AutoModelForCausalLM": "pGenomeOcean/GenomeOcean-100M--modeling_mistral.MistralForCausalLM",
11
+ "AutoModelForMaskedLM": "pGenomeOcean/GenomeOcean-100M--modeling_mistral.MistralForMaskedLM",
12
+ "AutoModelForSequenceClassification": "pGenomeOcean/GenomeOcean-100M--modeling_mistral.MistralForSequenceClassification"
13
+ },
14
+ "bos_token_id": 1,
15
+ "classifier_dropout": 0.1,
16
+ "eos_token_id": 2,
17
+ "hidden_act": "silu",
18
+ "hidden_size": 768,
19
+ "initializer_range": 0.02,
20
+ "intermediate_size": 3072,
21
+ "is_causal": true,
22
+ "max_position_embeddings": 32768,
23
+ "model_type": "mistral",
24
+ "num_attention_heads": 8,
25
+ "num_hidden_layers": 12,
26
+ "num_key_value_heads": 8,
27
+ "output_router_logits": false,
28
+ "pad_token_id": 3,
29
+ "problem_type": "single_label_classification",
30
+ "rms_norm_eps": 1e-05,
31
+ "rope_theta": 1000000.0,
32
+ "router_aux_loss_coef": 0.02,
33
+ "sliding_window": null,
34
+ "tie_word_embeddings": false,
35
+ "torch_dtype": "float32",
36
+ "transformers_version": "4.45.2",
37
+ "use_cache": true,
38
+ "vocab_size": 4096
39
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c4f73e4913ffca2984a45bddfbf3ca91cc295d72b68c0e732d80648366342df
3
+ size 465662960
optimizer_state_dict.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cbbd8b9626feb859b7c53b9fd1aeeadd487b6303e94fd6316eb5fc5d1635f9d
3
+ size 931398901
special_tokens_map.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": {
3
+ "content": "[CLS]",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "mask_token": {
10
+ "content": "[MASK]",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "[PAD]",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "sep_token": {
24
+ "content": "[SEP]",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "unk_token": {
31
+ "content": "[UNK]",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ }
37
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[UNK]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "[CLS]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[SEP]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[PAD]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "mask_token": "[MASK]",
47
+ "model_max_length": 256,
48
+ "pad_token": "[PAD]",
49
+ "padding_side": "right",
50
+ "sep_token": "[SEP]",
51
+ "tokenizer_class": "PreTrainedTokenizerFast",
52
+ "unk_token": "[UNK]"
53
+ }
trainer_state.json ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.5273802876472473,
3
+ "best_model_checkpoint": "output_pipe/prom_300_tata/origin/checkpoint-200",
4
+ "epoch": 4.0,
5
+ "eval_steps": 200,
6
+ "global_step": 308,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.2987012987012987,
13
+ "grad_norm": 16.239261627197266,
14
+ "learning_rate": 2.4767441860465116e-05,
15
+ "loss": 0.6103,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 2.5974025974025974,
20
+ "grad_norm": 17.608713150024414,
21
+ "learning_rate": 1.313953488372093e-05,
22
+ "loss": 0.3571,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 2.5974025974025974,
27
+ "eval_accuracy": 0.7748776508972267,
28
+ "eval_f1": 0.7730506728478527,
29
+ "eval_loss": 0.5273802876472473,
30
+ "eval_matthews_correlation": 0.5791078000958941,
31
+ "eval_precision": 0.7973376111673984,
32
+ "eval_recall": 0.7819739511049428,
33
+ "eval_runtime": 0.1894,
34
+ "eval_samples_per_second": 3236.113,
35
+ "eval_steps_per_second": 52.791,
36
+ "step": 200
37
+ },
38
+ {
39
+ "epoch": 3.896103896103896,
40
+ "grad_norm": 13.628073692321777,
41
+ "learning_rate": 1.5116279069767443e-06,
42
+ "loss": 0.1742,
43
+ "step": 300
44
+ },
45
+ {
46
+ "epoch": 4.0,
47
+ "step": 308,
48
+ "total_flos": 946505703112704.0,
49
+ "train_loss": 0.37393945765185665,
50
+ "train_runtime": 21.4272,
51
+ "train_samples_per_second": 915.473,
52
+ "train_steps_per_second": 14.374
53
+ }
54
+ ],
55
+ "logging_steps": 100,
56
+ "max_steps": 308,
57
+ "num_input_tokens_seen": 0,
58
+ "num_train_epochs": 4,
59
+ "save_steps": 200,
60
+ "stateful_callbacks": {
61
+ "TrainerControl": {
62
+ "args": {
63
+ "should_epoch_stop": false,
64
+ "should_evaluate": false,
65
+ "should_log": false,
66
+ "should_save": true,
67
+ "should_training_stop": true
68
+ },
69
+ "attributes": {}
70
+ }
71
+ },
72
+ "total_flos": 946505703112704.0,
73
+ "train_batch_size": 64,
74
+ "trial_name": null,
75
+ "trial_params": null
76
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:927a7bf0cf20defe98619d51419261abf4c2ba74b85631d0fa7a781765676e42
3
+ size 5368