Update scripts/finetune.py
Browse filesCo-authored-by: NanoCode012 <[email protected]>
- scripts/finetune.py +0 -1
scripts/finetune.py
CHANGED
|
@@ -261,7 +261,6 @@ def train(
|
|
| 261 |
model.save_pretrained(cfg.output_dir)
|
| 262 |
return
|
| 263 |
|
| 264 |
-
model.train()
|
| 265 |
|
| 266 |
trainer = setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer)
|
| 267 |
|
|
|
|
| 261 |
model.save_pretrained(cfg.output_dir)
|
| 262 |
return
|
| 263 |
|
|
|
|
| 264 |
|
| 265 |
trainer = setup_trainer(cfg, train_dataset, eval_dataset, model, tokenizer)
|
| 266 |
|