FPN_Pytorch icon indicating copy to clipboard operation
FPN_Pytorch copied to clipboard

训练0712的时候显存变大

Open Johnccl opened this issue 7 years ago • 1 comments

@guoruoqian 请问为何在训练voc 0712数据集的时候显存占用会远远大于训练voc 07呢?我只能把batch size设为1才能训练。

Johnccl avatar Sep 04 '18 14:09 Johnccl

因为训练voc 07 时,为了节省显存,会把图片crop(保留中心区域)进行训练。而voc 0712数据中,有一张图片的长宽比很大(2.+),而图片中的物体只出现在图片两侧,如果把图片crop保留中心位置,这张图片就没有物体了,训练会报错,所以在训练voc 0712数据时,不使用crop操作,没有节省现存的行为。

ruoqianguo avatar Sep 05 '18 02:09 ruoqianguo