Internal: Blas GEMM launch failed : a.shape=(4096, 128), b.shape=(128, 312), m=4096, n=312, k=128

brightmart / albert_zh

A LITE BERT FOR SELF-SUPERVISED LEARNING OF LANGUAGE REPRESENTATIONS, 海量中文预训练ALBERT模型

https://arxiv.org/pdf/1909.11942.pdf

3.92k stars 755 forks source link

Internal: Blas GEMM launch failed : a.shape=(4096, 128), b.shape=(128, 312), m=4096, n=312, k=128 #178

Open wzhslap opened 11 months ago

wzhslap commented 11 months ago

Internal: Blas GEMM launch failed : a.shape=(4096, 128), b.shape=(128, 312), m=4096, n=312, k=128 [[node bert/embeddings/MatMul (defined at /albert_zh-master/modeling.py:523) ]]

why this happen? python 3.6 TensorFlow-gpu 1.14.0