Internal: Blas GEMM launch failed : a.shape=(4096, 128), b.shape=(128, 312), m=4096, n=312, k=128 #178
Description
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