mod
This commit is contained in:
parent
995b2f6164
commit
df542c0e6b
2
hft.py
2
hft.py
|
|
@ -171,7 +171,7 @@ training_args = TrainingArguments(
|
||||||
gradient_accumulation_steps=4,
|
gradient_accumulation_steps=4,
|
||||||
learning_rate=2e-5,
|
learning_rate=2e-5,
|
||||||
fp16=True,
|
fp16=True,
|
||||||
logging_steps=50, # Częstsze logowanie
|
logging_steps=1, # Częstsze logowanie
|
||||||
logging_dir="./logs", # Katalog na logi
|
logging_dir="./logs", # Katalog na logi
|
||||||
save_strategy="steps",
|
save_strategy="steps",
|
||||||
save_steps=1000,
|
save_steps=1000,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue