Skip to content

Commit

Permalink
fix(transformer): delete tmp attr
Browse files Browse the repository at this point in the history
  • Loading branch information
xiong-ma committed May 10, 2020
1 parent e5756b3 commit 482ed87
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions jina/executors/encoders/nlp/transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def encode(self, data: 'np.ndarray', *args, **kwargs) -> 'np.ndarray':
_mask_ids_batch = self.tensor2array(mask_ids_batch)
_seq_output = self.tensor2array(seq_output)
if self.pooling_strategy == 'cls':
if self._has_cls_token:
if hasattr(self._tokenizer, 'cls_token'):
output = self.tensor2array(extra_output[0])
else:
output = reduce_cls(_seq_output, _mask_ids_batch, self.cls_pos)
Expand Down Expand Up @@ -94,9 +94,6 @@ def post_init(self):
self._sess_func = None
self.tmp_model_path = self.model_abspath if os.path.exists(self.model_abspath) else self.model_name
self._tokenizer = self.get_tokenizer()

self._has_cls_token = hasattr(self._tokenizer, 'cls_token')

self.cls_pos = 'tail' if self.model_name == 'xlnet-base-cased' else 'head'

def array2tensor(self, array):
Expand Down

0 comments on commit 482ed87

Please sign in to comment.