Skip to content
This repository was archived by the owner on Feb 22, 2020. It is now read-only.

Commit fbfa1e4

Browse files
committed
fix(transformer): add model eval
1 parent 2066beb commit fbfa1e4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

gnes/encoder/text/transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def post_init(self):
5050
(RobertaModel, RobertaTokenizer, 'roberta-base')]}[self.model_name]
5151

5252
def load_model_tokenizer(x):
53-
return model_class.from_pretrained(x), tokenizer_class.from_pretrained(x)
53+
return model_class.from_pretrained(x).eval(), tokenizer_class.from_pretrained(x)
5454

5555
try:
5656
self.model, self.tokenizer = load_model_tokenizer(self.work_dir)

0 commit comments

Comments
 (0)