Can this be used for MultiClass classification or only for 2?
I am running into a  this error:
[CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling cublasCreate(handle)]
and have found that this may be an issue due to an inconsistency between number of labels and number of output units.