原文:RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)

训练arcface时报这个错误。 原来是因为类别数没有设置正确。 ...

2020-08-11 10:46 0 8618 推荐指数:

查看详情

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM