mod
This commit is contained in:
parent
b3d1e0295a
commit
90f3d66723
1
hft.py
1
hft.py
|
|
@ -170,7 +170,6 @@ model.gradient_checkpointing_enable()
|
||||||
training_args = TrainingArguments(
|
training_args = TrainingArguments(
|
||||||
output_dir="./results",
|
output_dir="./results",
|
||||||
num_train_epochs=3,
|
num_train_epochs=3,
|
||||||
per_device_train_batch_size=2,
|
|
||||||
gradient_accumulation_steps=4,
|
gradient_accumulation_steps=4,
|
||||||
learning_rate=2e-5,
|
learning_rate=2e-5,
|
||||||
fp16=True,
|
fp16=True,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue