AjayP13 commited on
Commit
3790910
1 Parent(s): a23d134

Pushed by DataDreamer

Browse files

Update training_args.json

Files changed (1) hide show
  1. 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
+ }