一、实验设置
自己的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 版权协议,转载请附上原文出处链接和本声明。