本文新增添分类，检测，换脸技术等学习教程，各种调参技巧和tricks，卷积结构详细解析可视化，注意力机制代码等详解！本次垃圾分类挑战杯，目的在于构建基于深度学习技术的图像分类模型，实现垃圾图片类别的精准识别，大赛参考深圳垃圾分类标准，按可回收物、厨余垃圾、有害垃圾和其他垃圾四项分类。本项目包含完整的分类网络、数据增强、SVM等各种分类增强策略，后续还会继续更新新的分类技巧。
676
stars
176
forks
source link
你好，请问训练需要多大内存？我把batchsize设为10还是会OOM #8
Open
BeAShaper opened 5 years ago
@BeAShaper 你好这个是根据网络的结构来的，网络参数越多batch设置的就越小。我的2个是1080Ti，B4网络batch是24 B5是16