2020年8月11日

RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)
摘要： 训练arcface时报这个错误。 原来是因为类别数没有设置正确。 阅读全文
posted @ 2020-08-11 10:46 兔子二 阅读(14620) 评论(0) 推荐(0) 编辑

导航