一、实验设置
自己的dataset进行十分类实验，源代码的实验已成功跑通
二、bug
找到一个合适的代码之后，切换到自己的dataset，当图片格式完全匹配之后，报错：
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling ‘cublasCreate(handle)’
三、解决过程
1.batch_size过大，memory占用太多–>调小batch_size到4，仍然行不通
2.在调用模型的时候没有model.cuda（）–>加上之后，行不通，仍然报错
3.stack overflow上相似错误显示是数据集标签的对应问题–>注意到自己的标签是从1-10，而应该是0-9，修改之后，成功！
版权声明：本文为MarsandOcean原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。