diff options
| author | ZJaume <jzaragoza@prompsit.com> | 2022-07-20 16:21:33 +0300 |
|---|---|---|
| committer | Jaume Zaragoza <ZJaume@users.noreply.github.com> | 2022-07-27 15:20:55 +0300 |
| commit | 7d241321c3a7f40631694329f30a11017ea15457 (patch) | |
| tree | b0be8024a27f381ec6ab27947563a68438d544f1 | |
| parent | 3fbb1d3d122b3ee0d5df6f7b7eda89dd0b006b66 (diff) | |
Flatten the full model directory
| -rw-r--r-- | bicleaner_ai/models.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/bicleaner_ai/models.py b/bicleaner_ai/models.py index ea00bca..75dd80a 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -587,8 +587,6 @@ class BCXLMRoberta(BaseModel): shuffle=False) dev_generator.load(dev_set) - model_filename = self.dir + '/' + self.settings["model_file"] - vocab_filename = self.dir + '/' + self.settings["vocab_file"] earlystop = EarlyStopping(monitor='val_f1', mode='max', patience=self.settings["patience"], @@ -621,8 +619,8 @@ class BCXLMRoberta(BaseModel): batch_size=self.settings["batch_size"], callbacks=[earlystop], verbose=verbose) - self.model.save_pretrained(model_filename) - self.tokenizer.save_pretrained(vocab_filename) + self.model.save_pretrained(self.dir) + self.tokenizer.save_pretrained(self.dir) y_true = dev_generator.y with redirect_stdout(sys.stderr): |