关于训练epoch数

[galaxyGGG]

请问对于dota数据集训练s2anet来说，一般训练12个epoch就够了么？还是说需要训练上百甚至几百个epoch。实际上我在训练自己的遥感数据集时，loss在epoch=120、200时，loss大概能从0.08降到0.06左右，但是mAP相差不多，甚至更低了，请问一般应该如何选择训练epoch数呢？

另外，请问对于类别分布不均的情况，有的几千有的只有几十个，可以使用什么方法解决呢？谢谢！

[csuhan]

epoch数量：取决于不同的数据集，一般采用n*12 epoch设置，n取值需要实验。 类别不均衡：对少样本类重采样/数据增强，增加样本数量。

[busbyjrj]

您好，请问一下采用n12 epoch是出于什么训练上的目的，或者是遵守某种约定么？似乎有一半左右的DOTA的论文采用着n12 epoch的设定，也有一些将epoch设置到100、200...，这样训练epoch相差很多了。谢谢！

[csuhan]

@busbyjrj 12 epoch=1X schedule, 基本都是follow detectron的。 100-200 epoch一般被centernet、detr等模型采用。