AngelaGijon commited on
Commit
2968be9
verified
1 Parent(s): 8ae7a59

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -24
README.md CHANGED
@@ -40,28 +40,28 @@ La Configuraci贸n 3, con un n煤mero de 茅pocas superior, logr贸 la convergencia
40
 
41
  ## Argumentos de entrenamiento
42
 
43
- training_args = Seq2SeqTrainingArguments(
44
- output_dir=output_dir,
45
- lr_scheduler_type="linear",
46
- num_train_epochs=15,
47
- per_device_train_batch_size=8,
48
- per_device_eval_batch_size=8,
49
- gradient_accumulation_steps=2,
50
- learning_rate=5e-5,
51
- weight_decay=0.01,
52
- warmup_steps=50,
53
- predict_with_generate=True,
54
- logging_dir=f"{output_dir}/logs",
55
- logging_steps=10,
56
- eval_steps=10,
57
- save_total_limit=2,
58
- dataloader_num_workers=2,
59
- evaluation_strategy="steps",
60
- save_strategy="steps",
61
- load_best_model_at_end=True,
62
- fp16=True,
63
- seed=42,
64
- )
65
 
66
  ## Dataset
67
 
@@ -83,9 +83,11 @@ Splits disponibles:
83
 
84
  Puedes cargar el dataset directamente desde Hugging Face:
85
 
86
- _from datasets import load_dataset
 
87
  ds_train = load_dataset("AngelaGijon/TFG-DockerCommands-Dataset", split="train")
88
- print(ds_train[0])_
 
89
 
90
  Disponible en el repositorio: **"AngelaGijon/TFG-DockerCommands-Dataset"**
91
 
 
40
 
41
  ## Argumentos de entrenamiento
42
 
43
+ | Argumento | Valor |
44
+ | ----------------------------- | ---------------------: |
45
+ | `output_dir` | `output_dir` |
46
+ | `lr_scheduler_type` | `"linear"` |
47
+ | `num_train_epochs` | `15` |
48
+ | `per_device_train_batch_size` | `8` |
49
+ | `per_device_eval_batch_size` | `8` |
50
+ | `gradient_accumulation_steps` | `2` |
51
+ | `learning_rate` | `5e-5` |
52
+ | `weight_decay` | `0.01` |
53
+ | `warmup_steps` | `50` |
54
+ | `predict_with_generate` | `True` |
55
+ | `logging_dir` | `f"{output_dir}/logs"` |
56
+ | `logging_steps` | `10` |
57
+ | `eval_steps` | `10` |
58
+ | `save_total_limit` | `2` |
59
+ | `dataloader_num_workers` | `2` |
60
+ | `evaluation_strategy` | `"steps"` |
61
+ | `save_strategy` | `"steps"` |
62
+ | `load_best_model_at_end` | `True` |
63
+ | `fp16` | `True` |
64
+ | `seed` | `42` |
65
 
66
  ## Dataset
67
 
 
83
 
84
  Puedes cargar el dataset directamente desde Hugging Face:
85
 
86
+ ```
87
+ from datasets import load_dataset
88
  ds_train = load_dataset("AngelaGijon/TFG-DockerCommands-Dataset", split="train")
89
+ print(ds_train[0])
90
+ ```
91
 
92
  Disponible en el repositorio: **"AngelaGijon/TFG-DockerCommands-Dataset"**
93