原文: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