vrd-dsr
vrd-dsr copied to clipboard
Changing the batch size
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.
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.