xmxoxo / BERT-train2deploy

BERT模型从训练到部署
517 stars 165 forks source link

使用TensorFlow-gpu会报错,但使用TensorFlow-cpu不会 #30

Open sweetboxwwy opened 2 years ago

sweetboxwwy commented 2 years ago

InternalError (see above for traceback): Blas GEMM launch failed : a.shape=(1024, 2), b.shape=(2, 768), m=1024, n=768, k=2 [[node bert/embeddings/MatMul (defined at D:\PycharmProjects\GitHubProjects\BERT-train2deploy-master\BERT-train2deploy-master\modeling.py:486) ]] [[node mean/broadcast_weights/assert_broadcastable/is_valid_shape/has_valid_nonscalar_shape/has_invalid_dims/concat (defined at D:/PycharmProjects/GitHubProjects/BERT-train2deploy-master/BERT-train2deploy-master/run_mobile.py:756) ]]