Skip to content
This repository has been archived by the owner. It is now read-only.
Browse files
fix(transformer): add model eval
  • Loading branch information
jemmyshin committed Sep 6, 2019
1 parent 2066beb commit fbfa1e4746b42a75f807d5e691249048ccf4e732
Showing with 1 addition and 1 deletion.
  1. +1 −1 gnes/encoder/text/
@@ -50,7 +50,7 @@ def post_init(self):
(RobertaModel, RobertaTokenizer, 'roberta-base')]}[self.model_name]

def load_model_tokenizer(x):
return model_class.from_pretrained(x), tokenizer_class.from_pretrained(x)
return model_class.from_pretrained(x).eval(), tokenizer_class.from_pretrained(x)

self.model, self.tokenizer = load_model_tokenizer(self.work_dir)

0 comments on commit fbfa1e4

Please sign in to comment.