pytorch训练神经网络,出现下面错误。
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb
label数量和最后一层分类器不匹配。
(lll¬ω¬)
pytorch训练神经网络,出现下面错误。
RuntimeError: CUDA error: CUBLAS_STATUS_EXECUTION_FAILED when calling `cublasSgemm( handle, opa, opb
label数量和最后一层分类器不匹配。
(lll¬ω¬)