This commit is contained in:
l.gabrysiak 2025-02-26 11:17:48 +01:00
parent 84d7926e9e
commit f826dc05ca
1 changed files with 13 additions and 13 deletions

26
gpt.py
View File

@ -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