Pushed by DataDreamer
Browse filesUpdate training_args.json
- training_args.json +129 -0
training_args.json
ADDED
@@ -0,0 +1,129 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"output_dir": "./output/train-an-abstract--tweet-model/_checkpoints",
|
3 |
+
"overwrite_output_dir": false,
|
4 |
+
"do_train": false,
|
5 |
+
"do_eval": true,
|
6 |
+
"do_predict": false,
|
7 |
+
"evaluation_strategy": "epoch",
|
8 |
+
"prediction_loss_only": false,
|
9 |
+
"per_device_train_batch_size": 8,
|
10 |
+
"per_device_eval_batch_size": 8,
|
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 |
+
"learning_rate": 0.001,
|
17 |
+
"weight_decay": 0.01,
|
18 |
+
"adam_beta1": 0.9,
|
19 |
+
"adam_beta2": 0.999,
|
20 |
+
"adam_epsilon": 1e-08,
|
21 |
+
"max_grad_norm": 1.0,
|
22 |
+
"num_train_epochs": 30,
|
23 |
+
"max_steps": -1,
|
24 |
+
"lr_scheduler_type": "linear",
|
25 |
+
"lr_scheduler_kwargs": {},
|
26 |
+
"warmup_ratio": 0.0,
|
27 |
+
"warmup_steps": 0,
|
28 |
+
"log_level": "passive",
|
29 |
+
"log_level_replica": "warning",
|
30 |
+
"log_on_each_node": true,
|
31 |
+
"logging_dir": "./output/train-an-abstract--tweet-model/_checkpoints/runs/Jan18_19-00-10_298e4036ac9d",
|
32 |
+
"logging_strategy": "steps",
|
33 |
+
"logging_first_step": false,
|
34 |
+
"logging_steps": 1,
|
35 |
+
"logging_nan_inf_filter": true,
|
36 |
+
"save_strategy": "epoch",
|
37 |
+
"save_steps": 500,
|
38 |
+
"save_total_limit": 1,
|
39 |
+
"save_safetensors": true,
|
40 |
+
"save_on_each_node": false,
|
41 |
+
"save_only_model": false,
|
42 |
+
"no_cuda": false,
|
43 |
+
"use_cpu": false,
|
44 |
+
"use_mps_device": false,
|
45 |
+
"seed": 42,
|
46 |
+
"data_seed": null,
|
47 |
+
"jit_mode_eval": false,
|
48 |
+
"use_ipex": false,
|
49 |
+
"bf16": false,
|
50 |
+
"fp16": false,
|
51 |
+
"fp16_opt_level": "O1",
|
52 |
+
"half_precision_backend": "auto",
|
53 |
+
"bf16_full_eval": false,
|
54 |
+
"fp16_full_eval": false,
|
55 |
+
"tf32": null,
|
56 |
+
"local_rank": 0,
|
57 |
+
"ddp_backend": null,
|
58 |
+
"tpu_num_cores": null,
|
59 |
+
"tpu_metrics_debug": false,
|
60 |
+
"debug": [],
|
61 |
+
"dataloader_drop_last": false,
|
62 |
+
"eval_steps": null,
|
63 |
+
"dataloader_num_workers": 0,
|
64 |
+
"past_index": -1,
|
65 |
+
"run_name": "DataDreamer - Train an Abstract => Tweet Model",
|
66 |
+
"disable_tqdm": true,
|
67 |
+
"remove_unused_columns": true,
|
68 |
+
"label_names": null,
|
69 |
+
"load_best_model_at_end": true,
|
70 |
+
"metric_for_best_model": "eval_perplexity",
|
71 |
+
"greater_is_better": false,
|
72 |
+
"ignore_data_skip": false,
|
73 |
+
"fsdp": [],
|
74 |
+
"fsdp_min_num_params": 0,
|
75 |
+
"fsdp_config": {
|
76 |
+
"min_num_params": 0,
|
77 |
+
"xla": false,
|
78 |
+
"xla_fsdp_grad_ckpt": false
|
79 |
+
},
|
80 |
+
"fsdp_transformer_layer_cls_to_wrap": null,
|
81 |
+
"deepspeed": null,
|
82 |
+
"label_smoothing_factor": 0.0,
|
83 |
+
"optim": "adamw_torch",
|
84 |
+
"optim_args": null,
|
85 |
+
"adafactor": false,
|
86 |
+
"group_by_length": false,
|
87 |
+
"length_column_name": "length",
|
88 |
+
"report_to": [],
|
89 |
+
"ddp_find_unused_parameters": null,
|
90 |
+
"ddp_bucket_cap_mb": null,
|
91 |
+
"ddp_broadcast_buffers": null,
|
92 |
+
"dataloader_pin_memory": true,
|
93 |
+
"dataloader_persistent_workers": false,
|
94 |
+
"skip_memory_metrics": true,
|
95 |
+
"use_legacy_prediction_loop": false,
|
96 |
+
"push_to_hub": false,
|
97 |
+
"resume_from_checkpoint": null,
|
98 |
+
"hub_model_id": null,
|
99 |
+
"hub_strategy": "every_save",
|
100 |
+
"hub_token": "<HUB_TOKEN>",
|
101 |
+
"hub_private_repo": false,
|
102 |
+
"hub_always_push": false,
|
103 |
+
"gradient_checkpointing": false,
|
104 |
+
"gradient_checkpointing_kwargs": null,
|
105 |
+
"include_inputs_for_metrics": false,
|
106 |
+
"fp16_backend": "auto",
|
107 |
+
"push_to_hub_model_id": null,
|
108 |
+
"push_to_hub_organization": null,
|
109 |
+
"push_to_hub_token": "<PUSH_TO_HUB_TOKEN>",
|
110 |
+
"mp_parameters": "",
|
111 |
+
"auto_find_batch_size": false,
|
112 |
+
"full_determinism": false,
|
113 |
+
"torchdynamo": null,
|
114 |
+
"ray_scope": "last",
|
115 |
+
"ddp_timeout": 1800,
|
116 |
+
"torch_compile": false,
|
117 |
+
"torch_compile_backend": null,
|
118 |
+
"torch_compile_mode": null,
|
119 |
+
"dispatch_batches": null,
|
120 |
+
"split_batches": false,
|
121 |
+
"include_tokens_per_second": false,
|
122 |
+
"include_num_input_tokens_seen": false,
|
123 |
+
"neftune_noise_alpha": null,
|
124 |
+
"sortish_sampler": false,
|
125 |
+
"predict_with_generate": false,
|
126 |
+
"generation_max_length": null,
|
127 |
+
"generation_num_beams": null,
|
128 |
+
"generation_config": null
|
129 |
+
}
|