csharma commented on issue #16721: GPU is not enabled for mxnet based word
embeddings.
URL:
https://github.com/apache/incubator-mxnet/issues/16721#issuecomment-549549560
Add the following line to make it run
from bert_embedding import BertEmbedding
Best,
Cartik
--
csharma commented on issue #16721: GPU is not enabled for mxnet based word
embeddings.
URL:
https://github.com/apache/incubator-mxnet/issues/16721#issuecomment-549541198
Add this line,
from bert_embedding import BertEmbedding
Yes, the laptop I am using has GPU support with 2