Set model to train mode for finetuning (#1525)

Signed-off-by: yiliu30 <[email protected]>
intel · Jan 11, 2024 · 061884d · 061884d
1 parent d96e14a
commit 061884d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/...ples/pytorch/nlp/huggingface_models/question-answering/pruning/eager/run_qa_no_trainer.py b/...ples/pytorch/nlp/huggingface_models/question-answering/pruning/eager/run_qa_no_trainer.py
@@ -892,6 +892,7 @@ def create_and_fill_np_array(start_or_end_logits, dataset, max_len):
         teacher_model.eval()
     else:
         # Prepare everything with our `accelerator`.
+        model.train()
         model, optimizer, train_dataloader, eval_dataloader, lr_scheduler = accelerator.prepare(
             model, optimizer, train_dataloader, eval_dataloader, lr_scheduler
         )