extract_features.py is too slow

google-research / bert

TensorFlow code and pre-trained models for BERT

https://arxiv.org/abs/1810.04805

Apache License 2.0

38.21k stars 9.61k forks source link

extract_features.py is too slow #828

Open sjm1992st opened 5 years ago

sjm1992st commented 5 years ago

when i used the code in extract_features.py ''' for result in self.estimator.predict(input_fn, yield_single_examples=True) ''' i found it spend about 4s，could some ways can i do？

AmeeraMilibari commented 4 years ago

@sjm1992st check this, it may help! https://github.com/hanxiao/bert-as-service