- 博客(2)
- 资源 (3)
- 收藏
- 关注
RSS订阅原创 关于Python报错：RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCre`
在网上查了很多方案，在Torch社区也找了不少方案，但都对不上。后来觉着这是CUDA的封装问题，将真正的错误信息掩盖了。于是将程序改成在CPU上跑。报错信息变成标签错误，进而发现是数据预处理打标签时错误。修改相关代码后成功运行。...
2021-07-16 11:27:44
1611
1
转载 Torch transforms 数据预处理
转自：https://blog.csdn.net/pengchengliu/article/details/108668793 一、数据增强概述 二、数据增强方法：裁剪 三、数据增强方法：翻转和旋转 四、数据增强方法：变换 五、transforms方法的选择操作 一、数据增强概述 我们来看图片中的数据增强是怎么样的。 左边的图像可以增强得到右边的图片，供模型训练。模型见过更多更丰富的样本，从
2021-04-02 11:45:34
1437
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人