mod gpt epochs = 8

This commit is contained in:
l.gabrysiak 2025-02-26 10:07:09 +01:00
parent 2a4404d708
commit 6816562163
1 changed files with 1 additions and 1 deletions

2
gpt.py
View File

@ -81,7 +81,7 @@ def main():
# Konfiguracja treningu # Konfiguracja treningu
training_args = TrainingArguments( training_args = TrainingArguments(
output_dir="./results", output_dir="./results",
num_train_epochs=5, # Zwiększono liczbę epok num_train_epochs=8, # Zwiększono liczbę epok
per_device_train_batch_size=2, per_device_train_batch_size=2,
learning_rate=5e-5, learning_rate=5e-5,
logging_steps=10, logging_steps=10,