update save model.

shibing624 · Feb 13, 2022 · a26aac5 · a26aac5
1 parent 285da40
commit a26aac5
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/autocomplete/gpt2_coder.py b/autocomplete/gpt2_coder.py
@@ -131,8 +131,6 @@ def train_model(
             max_grad_norm=max_grad_norm,
             max_steps=max_steps
         )
-
-        self.save_model(output_dir, model=self.model)
         logger.info(f" Training of GPT2 model complete. Saved to {output_dir}.")
 
         return global_step, training_details