--gpu自然是选择设备上的gpu核心,以目前我的认知,训练过程只能选择一个gpu核心,所以需要的显存不是一般电脑可以接受的;
--solver这是一个控制训练过程或者说设置训练过程中的一些参数的文件,它不决定网络的参数,但是它决定了学习率,动量值,以及其它一些在训练过程中可以调整的参数,具体会在学习solver的时候再整理;--iters就是训练的轮数,faster-rcnn每轮的训练都是选取两张图片并从中选取128(每张64)个bounding box进行网络的训练,当然其中包含有正样本也有随机选取的负样本,负样本与正样本的IOU应小于一个阈值;
--weights这是需要你给出预训练的模型,原理上看就是说很多有实力的机构用高级GPU且花费大量时间训练得到的模型(指模型中的参数,模型的结构当然是已经定好的),深度学习的参数太多,想从0开始训练就需要有足够的数据和足够的时间最重要的是要有足够强大的硬件支持,所以imagenet这个1.几T的高清图片数据库就为深度学习的训练提供了足够的数据,现有的大量工作都是在各种预训练好的模型基础上完成的;
---------------------
作者:o_ochao
来源:CSDN
原文:https://blog.csdn.net/o_ochao/article/details/51981950
版权声明:本文为博主原创文章,转载请附上博文链接!