配置文件voc总量是怎么换算
Created by: xiaomujiang
阅读文档的时候,看到训练集数量约为12万张(下面一段话),然后换算了epoch , 所以我想问一下,在其他的配置文件中,如果是voc格式的,原配置文件的训练集数量是多少??是voc02 还是12 还是加和,这个怎么看?
修改训练轮数与学习率等参数: 根据训练集数量与总batch_size大小计算epoch数,然后将epoch数换算得到训练总轮数max_iters。milestones(学习率变化界限)也是同理。原配置文件中总batch_size=2*8=16(8卡训练),训练集数量约为12万张,max_iters=90000,所以epoch数=16x90000/120000=12。在AI识虫数据集中,训练集数量约为1700,在单卡GPU上训练,max_iters=12x1700/2=10200。同理计算milestones为: [6800, 9000]。