https://blog.csdn.net/u014176855/article/details/80994991
他的方法对我这个不奏效。检查发现是分类网络的全连接layer第一层神经元的数量设置错误。
CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling `cublasSgemm( handle, opa, opb, m, n, k, &alpha
最新推荐文章于 2024-05-21 16:19:15 发布
https://blog.csdn.net/u014176855/article/details/80994991
他的方法对我这个不奏效。检查发现是分类网络的全连接layer第一层神经元的数量设置错误。