nielsbantilan commited on
Commit
398bb4c
·
1 Parent(s): 6e2dc66

Upload flyte_training_config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. flyte_training_config.json +1 -1
flyte_training_config.json CHANGED
@@ -1 +1 @@
1
- {"base_model": "EleutherAI/pythia-1B-deduped", "data_path": "wikipedia", "data_name": "20220301.simple", "num_epochs": 1, "max_steps": 100, "learning_rate": 2e-05, "weight_decay": 0.02, "warmup_ratio": 0.03, "lr_scheduler_type": "cosine", "batch_size": 8, "micro_batch_size": 1, "val_set_size": 0, "group_by_length": false, "instruction_key": "instruction", "input_key": "input", "output_key": "output", "device_map": "auto", "cache_dir": null, "optim": "adamw_torch", "model_max_length": 512, "debug_mode": false, "debug_train_data_size": 1024, "wandb_project": ""}
 
1
+ {"base_model": "EleutherAI/pythia-1B-deduped", "data_path": "wikipedia", "data_name": "20220301.simple", "num_epochs": 1, "max_steps": 10, "learning_rate": 2e-05, "weight_decay": 0.02, "warmup_ratio": 0.03, "lr_scheduler_type": "cosine", "batch_size": 8, "micro_batch_size": 1, "val_set_size": 0, "group_by_length": false, "instruction_key": "instruction", "input_key": "input", "output_key": "output", "device_map": "auto", "cache_dir": null, "optim": "adamw_torch", "model_max_length": 512, "debug_mode": false, "debug_train_data_size": 1024, "wandb_project": ""}