AjayP13 commited on
Commit
4397af1
1 Parent(s): ba4e5db

Pushed by DataDreamer

Browse files

Update training_args.json

Files changed (1) hide show
  1. training_args.json +133 -0
training_args.json ADDED
@@ -0,0 +1,133 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "./.datadreamer/train-a-fictional-story--dagger-json-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": 1,
10
+ "per_device_eval_batch_size": 1,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 32,
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": 500,
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": "./.datadreamer/train-a-fictional-story--dagger-json-model/_checkpoints/runs/Apr06_23-37-51_nlpgpu07.seas.upenn.edu",
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
+ "dataloader_prefetch_factor": null,
65
+ "past_index": -1,
66
+ "run_name": "DataDreamer - Train a Fictional Story => DAGGER JSON Model",
67
+ "disable_tqdm": true,
68
+ "remove_unused_columns": true,
69
+ "label_names": null,
70
+ "load_best_model_at_end": true,
71
+ "metric_for_best_model": "eval_perplexity",
72
+ "greater_is_better": false,
73
+ "ignore_data_skip": false,
74
+ "fsdp": [],
75
+ "fsdp_min_num_params": 0,
76
+ "fsdp_config": {
77
+ "min_num_params": 0,
78
+ "xla": false,
79
+ "xla_fsdp_v2": false,
80
+ "xla_fsdp_grad_ckpt": false
81
+ },
82
+ "fsdp_transformer_layer_cls_to_wrap": null,
83
+ "accelerator_config": {
84
+ "split_batches": false,
85
+ "dispatch_batches": null,
86
+ "even_batches": true,
87
+ "use_seedable_sampler": true
88
+ },
89
+ "deepspeed": null,
90
+ "label_smoothing_factor": 0.0,
91
+ "optim": "adamw_torch",
92
+ "optim_args": null,
93
+ "adafactor": false,
94
+ "group_by_length": false,
95
+ "length_column_name": "length",
96
+ "report_to": [],
97
+ "ddp_find_unused_parameters": null,
98
+ "ddp_bucket_cap_mb": null,
99
+ "ddp_broadcast_buffers": null,
100
+ "dataloader_pin_memory": true,
101
+ "dataloader_persistent_workers": false,
102
+ "skip_memory_metrics": true,
103
+ "use_legacy_prediction_loop": false,
104
+ "push_to_hub": false,
105
+ "resume_from_checkpoint": null,
106
+ "hub_model_id": null,
107
+ "hub_strategy": "every_save",
108
+ "hub_token": "<HUB_TOKEN>",
109
+ "hub_private_repo": false,
110
+ "hub_always_push": false,
111
+ "gradient_checkpointing": false,
112
+ "gradient_checkpointing_kwargs": null,
113
+ "include_inputs_for_metrics": false,
114
+ "fp16_backend": "auto",
115
+ "push_to_hub_model_id": null,
116
+ "push_to_hub_organization": null,
117
+ "push_to_hub_token": "<PUSH_TO_HUB_TOKEN>",
118
+ "mp_parameters": "",
119
+ "auto_find_batch_size": false,
120
+ "full_determinism": false,
121
+ "torchdynamo": null,
122
+ "ray_scope": "last",
123
+ "ddp_timeout": 1800,
124
+ "torch_compile": false,
125
+ "torch_compile_backend": null,
126
+ "torch_compile_mode": null,
127
+ "dispatch_batches": null,
128
+ "split_batches": null,
129
+ "include_tokens_per_second": false,
130
+ "include_num_input_tokens_seen": false,
131
+ "neftune_noise_alpha": null,
132
+ "optim_target_modules": null
133
+ }