github
/
ChatGLM2-6B
mirror of https://github.com/THUDM/ChatGLM2-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add train epochs

pull/672/head
wanghh2000 2023-12-29 15:15:09 +08:00
parent e1b760b706
commit d4048deebf
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
finetune/finetune-lora.py
View File

@ -47,6 +47,7 @@ trainer = Trainer(
model=lora_model, model=lora_model,
train_dataset=train_ds, train_dataset=train_ds,
args=TrainingArguments( args=TrainingArguments(
num_train_epochs=1,
per_device_train_batch_size=4, per_device_train_batch_size=4,
gradient_accumulation_steps=4, gradient_accumulation_steps=4,
warmup_steps=3, warmup_steps=3,