From 8957e81c0e1c5719e817eea668084d117be2882c Mon Sep 17 00:00:00 2001 From: davebulaval Date: Sat, 9 Mar 2024 12:52:59 -0500 Subject: [PATCH] save best model saving scheme --- src/training/few_shot_training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/training/few_shot_training.py b/src/training/few_shot_training.py index 13d617f..1026c33 100644 --- 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}")