MobileNet-YOLO icon indicating copy to clipboard operation
MobileNet-YOLO copied to clipboard

train和test不一致问题

Open zpge opened this issue 4 years ago • 3 comments

发现一个问题,train的时候可以用很大的batchsize和image size,但是每次test的时候就要用很小的image size和batch size,不知道这是什么原因造成的呢?

test是指训练完N次后的validation。

zpge avatar Mar 04 '20 08:03 zpge

@zpge 解决了没,我也是在test的时候就out of memory了

sditeng avatar Mar 10 '20 09:03 sditeng

@zpge 解决了没,我也是在test的时候就out of memory了

没有解决,可能可以尝试把train和test分开。train的时候不test,后面单独尝试test得到mAP。我还没试过。

zpge avatar Mar 10 '20 10:03 zpge

如果设置了在训练过程中的test,将会建构另外的test网络进行阶段性验证,所以会用到更多的显存,如果显存不够,还是另行test

WangYX-TKZ avatar May 25 '20 07:05 WangYX-TKZ