save best model saving scheme

GRAAL-Research · Mar 9, 2024 · 8957e81 · 8957e81
1 parent 5414c58
commit 8957e81
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/training/few_shot_training.py b/src/training/few_shot_training.py
@@ -21,7 +21,7 @@
 log.propagate = False
 log.setLevel(logging.ERROR)
 
-num_epoch = 250
+num_epoch = 500
 
 parser = argparse.ArgumentParser()
 parser.add_argument(
@@ -124,4 +124,4 @@ def tokenize_function(example):
     metric_key_prefix="test/unrelated_sentences",
 )
 
-trainer.save_model(f"meaningbert_best_model")
+trainer.save_model(f"meaningbert_best_model_{seed}")