Plasmoxy commited on
Commit
3a38ec8
·
verified ·
1 Parent(s): 16df8d9

End of training

Browse files
README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/flan-t5-small
5
+ tags:
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: flan-t5-small-gigatrue-savetest
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # flan-t5-small-gigatrue-savetest
16
+
17
+ This model is a fine-tuned version of [google/flan-t5-small](https://huggingface.co/google/flan-t5-small) on an unknown dataset.
18
+
19
+ ## Model description
20
+
21
+ More information needed
22
+
23
+ ## Intended uses & limitations
24
+
25
+ More information needed
26
+
27
+ ## Training and evaluation data
28
+
29
+ More information needed
30
+
31
+ ## Training procedure
32
+
33
+ ### Training hyperparameters
34
+
35
+ The following hyperparameters were used during training:
36
+ - learning_rate: 0.0003
37
+ - train_batch_size: 256
38
+ - eval_batch_size: 256
39
+ - seed: 42
40
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
41
+ - lr_scheduler_type: linear
42
+ - num_epochs: 0.01
43
+
44
+ ### Training results
45
+
46
+
47
+
48
+ ### Framework versions
49
+
50
+ - Transformers 4.45.2
51
+ - Pytorch 2.5.1
52
+ - Datasets 3.2.0
53
+ - Tokenizers 0.20.3
config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/flan-t5-small",
3
+ "architectures": [
4
+ "T5ForConditionalGeneration"
5
+ ],
6
+ "classifier_dropout": 0.0,
7
+ "d_ff": 1024,
8
+ "d_kv": 64,
9
+ "d_model": 512,
10
+ "decoder_start_token_id": 0,
11
+ "dense_act_fn": "gelu_new",
12
+ "dropout_rate": 0.1,
13
+ "eos_token_id": 1,
14
+ "feed_forward_proj": "gated-gelu",
15
+ "initializer_factor": 1.0,
16
+ "is_encoder_decoder": true,
17
+ "is_gated_act": true,
18
+ "layer_norm_epsilon": 1e-06,
19
+ "model_type": "t5",
20
+ "n_positions": 512,
21
+ "num_decoder_layers": 3,
22
+ "num_heads": 6,
23
+ "num_layers": 8,
24
+ "output_past": true,
25
+ "pad_token_id": 0,
26
+ "relative_attention_max_distance": 128,
27
+ "relative_attention_num_buckets": 32,
28
+ "task_specific_params": {
29
+ "summarization": {
30
+ "early_stopping": true,
31
+ "length_penalty": 2.0,
32
+ "max_length": 200,
33
+ "min_length": 30,
34
+ "no_repeat_ngram_size": 3,
35
+ "num_beams": 4,
36
+ "prefix": "summarize: "
37
+ },
38
+ "translation_en_to_de": {
39
+ "early_stopping": true,
40
+ "max_length": 300,
41
+ "num_beams": 4,
42
+ "prefix": "translate English to German: "
43
+ },
44
+ "translation_en_to_fr": {
45
+ "early_stopping": true,
46
+ "max_length": 300,
47
+ "num_beams": 4,
48
+ "prefix": "translate English to French: "
49
+ },
50
+ "translation_en_to_ro": {
51
+ "early_stopping": true,
52
+ "max_length": 300,
53
+ "num_beams": 4,
54
+ "prefix": "translate English to Romanian: "
55
+ }
56
+ },
57
+ "tie_word_embeddings": false,
58
+ "torch_dtype": "bfloat16",
59
+ "transformers_version": "4.45.2",
60
+ "use_cache": true,
61
+ "vocab_size": 32128
62
+ }
coreconfig.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "out_name": "flan-t5-small-gigatrue-savetest",
3
+ "model_archetype": "flan-t5",
4
+ "model_name": "google/flan-t5-small",
5
+ "tokenizer_name": "google/flan-t5-small",
6
+ "model_torch_dtype": "bfloat16",
7
+ "dataset_lang": "en",
8
+ "dataset_name": "Plasmoxy/gigatrue",
9
+ "dataset_tokenized_cache_name": "gigatrue_tokenized_flan_t5_110-35",
10
+ "use_half_val_dataset": true,
11
+ "max_input_length": 110,
12
+ "max_target_length": 35,
13
+ "batch_size": 256,
14
+ "learning_rate": 0.0003,
15
+ "num_train_epochs": 0.01,
16
+ "pkg_versions": {
17
+ "optimum": "1.23.3",
18
+ "transformers": "4.45.2",
19
+ "openvino": "2024.6.0",
20
+ "nncf": "2.14.1",
21
+ "torch": "2.5.1",
22
+ "datasets": "3.2.0",
23
+ "peft": "0.13.3.dev0"
24
+ }
25
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "decoder_start_token_id": 0,
4
+ "eos_token_id": 1,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "4.45.2"
7
+ }
logs/events.out.tfevents.1735391585.gna4000.583374.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d0ce0b73d33162fc20bdc25ed7d8d3d6445d7efa9ff784580312f216dc6a9e86
3
+ size 6617
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4125e06ec577983de798e57dc7c8a61d35f3c048022afd71751828a1abf75c50
3
+ size 122463808
nvidia_smi_early.log ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Sat Dec 28 14:13:35 2024
2
+ +-----------------------------------------------------------------------------------------+
3
+ | NVIDIA-SMI 550.127.08 Driver Version: 550.127.08 CUDA Version: 12.4 |
4
+ |-----------------------------------------+------------------------+----------------------+
5
+ | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
6
+ | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
7
+ | | | MIG M. |
8
+ |=========================================+========================+======================|
9
+ | 0 NVIDIA RTX A4000 Off | 00000000:01:00.0 Off | Off |
10
+ | 41% 65C P2 137W / 140W | 12057MiB / 16376MiB | 100% Default |
11
+ | | | N/A |
12
+ +-----------------------------------------+------------------------+----------------------+
13
+
14
+ +-----------------------------------------------------------------------------------------+
15
+ | Processes: |
16
+ | GPU GI CI PID Type Process name GPU Memory |
17
+ | ID ID Usage |
18
+ |=========================================================================================|
19
+ | 0 N/A N/A 76833 G /usr/lib/xorg/Xorg 4MiB |
20
+ | 0 N/A N/A 583374 C ...ik/miniforge3/envs/alpha/bin/python 12042MiB |
21
+ +-----------------------------------------------------------------------------------------+
trainer_state.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.010012854340031121,
5
+ "eval_steps": 3000,
6
+ "global_step": 148,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 6.76544212164265e-05,
13
+ "grad_norm": 36.75,
14
+ "learning_rate": 0.00029797297297297294,
15
+ "loss": 9.0625,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.010012854340031121,
20
+ "step": 148,
21
+ "total_flos": 1140002213855232.0,
22
+ "train_loss": 5.680320945945946,
23
+ "train_runtime": 58.256,
24
+ "train_samples_per_second": 649.516,
25
+ "train_steps_per_second": 2.541
26
+ }
27
+ ],
28
+ "logging_steps": 3000,
29
+ "max_steps": 148,
30
+ "num_input_tokens_seen": 0,
31
+ "num_train_epochs": 1,
32
+ "save_steps": 3000,
33
+ "stateful_callbacks": {
34
+ "TrainerControl": {
35
+ "args": {
36
+ "should_epoch_stop": false,
37
+ "should_evaluate": false,
38
+ "should_log": false,
39
+ "should_save": true,
40
+ "should_training_stop": true
41
+ },
42
+ "attributes": {}
43
+ }
44
+ },
45
+ "total_flos": 1140002213855232.0,
46
+ "train_batch_size": 256,
47
+ "trial_name": null,
48
+ "trial_params": null
49
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0449e3d25cfa1e51c81168447edc018b549e3d26e27879cdd6806130035885fb
3
+ size 5368
training_args.json ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "out_models/flan-t5-small-gigatrue-savetest",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": true,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 256,
10
+ "per_device_eval_batch_size": 256,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 1,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 0.0003,
18
+ "weight_decay": 0.0,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.999,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 0.01,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "linear",
26
+ "lr_scheduler_kwargs": {},
27
+ "warmup_ratio": 0.0,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "out_models/flan-t5-small-gigatrue-savetest/logs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 3000,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 3000,
39
+ "save_total_limit": null,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": null,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": false,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": 0,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": [],
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 3000,
65
+ "dataloader_num_workers": 0,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "out_models/flan-t5-small-gigatrue-savetest",
69
+ "disable_tqdm": false,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": null,
74
+ "greater_is_better": null,
75
+ "ignore_data_skip": false,
76
+ "fsdp": [],
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": {
79
+ "min_num_params": 0,
80
+ "xla": false,
81
+ "xla_fsdp_v2": false,
82
+ "xla_fsdp_grad_ckpt": false
83
+ },
84
+ "fsdp_transformer_layer_cls_to_wrap": null,
85
+ "accelerator_config": "AcceleratorConfig",
86
+ "deepspeed": null,
87
+ "label_smoothing_factor": 0.0,
88
+ "optim": "adamw_torch",
89
+ "optim_args": null,
90
+ "adafactor": false,
91
+ "group_by_length": false,
92
+ "length_column_name": "length",
93
+ "report_to": [
94
+ "tensorboard"
95
+ ],
96
+ "ddp_find_unused_parameters": null,
97
+ "ddp_bucket_cap_mb": null,
98
+ "ddp_broadcast_buffers": null,
99
+ "dataloader_pin_memory": true,
100
+ "dataloader_persistent_workers": false,
101
+ "skip_memory_metrics": true,
102
+ "use_legacy_prediction_loop": false,
103
+ "push_to_hub": false,
104
+ "resume_from_checkpoint": null,
105
+ "hub_model_id": null,
106
+ "hub_strategy": "every_save",
107
+ "hub_token": null,
108
+ "hub_private_repo": false,
109
+ "hub_always_push": false,
110
+ "gradient_checkpointing": false,
111
+ "gradient_checkpointing_kwargs": null,
112
+ "include_inputs_for_metrics": false,
113
+ "eval_do_concat_batches": true,
114
+ "fp16_backend": "auto",
115
+ "evaluation_strategy": null,
116
+ "push_to_hub_model_id": null,
117
+ "push_to_hub_organization": null,
118
+ "push_to_hub_token": null,
119
+ "mp_parameters": "",
120
+ "auto_find_batch_size": false,
121
+ "full_determinism": false,
122
+ "torchdynamo": null,
123
+ "ray_scope": "last",
124
+ "ddp_timeout": 1800,
125
+ "torch_compile": false,
126
+ "torch_compile_backend": null,
127
+ "torch_compile_mode": null,
128
+ "dispatch_batches": null,
129
+ "split_batches": null,
130
+ "include_tokens_per_second": false,
131
+ "include_num_input_tokens_seen": false,
132
+ "neftune_noise_alpha": null,
133
+ "optim_target_modules": null,
134
+ "batch_eval_metrics": false,
135
+ "eval_on_start": false,
136
+ "use_liger_kernel": false,
137
+ "eval_use_gather_object": false,
138
+ "sortish_sampler": false,
139
+ "predict_with_generate": false,
140
+ "generation_max_length": null,
141
+ "generation_num_beams": null,
142
+ "generation_config": null,
143
+ "distributed_state": "PartialState",
144
+ "_n_gpu": 1,
145
+ "__cached__setup_devices": "device",
146
+ "deepspeed_plugin": null
147
+ }