Skip to content

Commit

Permalink
model_root prepended multiple times if model is reloaded
Browse files Browse the repository at this point in the history
  • Loading branch information
Waino committed Sep 17, 2018
1 parent 54d0c30 commit 4f185f8
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions onmt/translate/translation_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -267,12 +267,14 @@ def load(self):
mode = self.tokenizer_opt["mode"]
else:
mode = None
# load can be called multiple times: modify copy
tokenizer_params = dict(self.tokenizer_opt["params"])
for key, value in self.tokenizer_opt["params"].items():
if key.endswith("path"):
self.tokenizer_opt["params"][key] = os.path.join(
tokenizer_params[key] = os.path.join(
self.model_root, value)
tokenizer = pyonmttok.Tokenizer(mode,
**self.tokenizer_opt["params"])
**tokenizer_params)
self.tokenizer = tokenizer
else:
raise ValueError("Invalid value for tokenizer type")
Expand Down

0 comments on commit 4f185f8

Please sign in to comment.