RoiPooling with batch_size > 1
AlbertoSabater opened this issue · 1 comments
AlbertoSabater commented
Hi! According to with the comments, the RoiPooling layer works with input size (1, rows, cols, channels)
, does that mean that it only works with batch_size = 1
, or can it be higher ((batch_size, rows, cols, channels))
?
Thanks in advance
Runist commented
set it for(-1, self.num_rois, self.pool_size, self.pool_size, self.channels), right?I training to add mult-batch, but something wrong.