mod gpt
This commit is contained in:
parent
84d7926e9e
commit
f826dc05ca
26
gpt.py
26
gpt.py
|
|
@ -89,19 +89,19 @@ def main():
|
|||
|
||||
# Konfiguracja treningu
|
||||
training_args = TrainingArguments(
|
||||
output_dir="./results",
|
||||
num_train_epochs=30,
|
||||
per_device_train_batch_size=2,
|
||||
learning_rate=1e-5,
|
||||
weight_decay=0.01,
|
||||
logging_steps=10,
|
||||
save_steps=100,
|
||||
eval_steps=100,
|
||||
evaluation_strategy="steps",
|
||||
load_best_model_at_end=True,
|
||||
report_to="none",
|
||||
save_total_limit=2,
|
||||
)
|
||||
output_dir="./results",
|
||||
num_train_epochs=30,
|
||||
per_device_train_batch_size=2,
|
||||
learning_rate=1e-5,
|
||||
weight_decay=0.01,
|
||||
logging_steps=10,
|
||||
save_steps=100,
|
||||
eval_steps=100,
|
||||
evaluation_strategy="steps",
|
||||
load_best_model_at_end=True,
|
||||
report_to="none",
|
||||
save_total_limit=2,
|
||||
)
|
||||
|
||||
|
||||
# Trainer
|
||||
|
|
|
|||
Loading…
Reference in New Issue