diff options
| author | ZJaume <jzaragoza@prompsit.com> | 2022-07-20 16:21:33 +0300 |
|---|---|---|
| committer | ZJaume <jzaragoza@prompsit.com> | 2022-07-20 16:21:33 +0300 |
| commit | d317024946a81d699e1f4cf691b534c3b93e4323 (patch) | |
| tree | 4efea8f7a5b74cd3e7aed45dab6695fc88775249 | |
| parent | 59a8bac7516ac99c6ba41cb603b75ea340a1cc0f (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 dff8fcf..1cc367e 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -586,8 +586,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"], @@ -620,8 +618,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): |