vrd-dsr icon indicating copy to clipboard operation
vrd-dsr copied to clipboard

Changing the batch size

Open sukrutrao opened this issue 6 years ago • 1 comments

Is it possible to change the batch size while training? I see that there is a _batch_size variable in the VRDDataLayer class, but the only place it is used seems to be here for setting the number of epochs. While actually doing the forward pass, it seems that only one image is used.

Is this the case? Will training code have to be modified to allow a higher batch size?

Thank you.

sukrutrao avatar Jun 13 '18 06:06 sukrutrao

I think it's possible to use more data in one batch. You need to modify the codes related to roi pooling and the ranking loss.

GriffinLiang avatar Jun 16 '18 13:06 GriffinLiang