diff options
| author | ZJaume <jzaragoza@prompsit.com> | 2022-09-15 17:11:49 +0300 |
|---|---|---|
| committer | ZJaume <jzaragoza@prompsit.com> | 2022-09-15 17:11:49 +0300 |
| commit | 3dd57097af9b0b1e6076d6baf03c94a7a5704ebd (patch) | |
| tree | cd7416eb0639def2956396b61ba34e2d4d7ba172 | |
| parent | e686b0e342b7f16db2d5b7ddece883aba84ac7b9 (diff) | |
Flatten full model directory (again)
| -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 138f3c7..cce7a83 100644 --- a/bicleaner_ai/models.py +++ b/bicleaner_ai/models.py @@ -633,10 +633,8 @@ class BCXLMRoberta(BaseModel): batch_size=self.settings["batch_size"], callbacks=[earlystop], verbose=verbose) - self.model.save_pretrained(self.dir + '/' - + self.settings["model_file"]) - self.tokenizer.save_pretrained(self.dir + '/' - + self.settings["vocab_file"]) + self.model.save_pretrained(self.dir) + self.tokenizer.save_pretrained(self.dir) y_true = dev_generator.y with redirect_stdout(sys.stderr): |