votenet icon indicating copy to clipboard operation
votenet copied to clipboard

Question on reproducing results on SUNRGBD data

Open StarsMyDestination opened this issue 5 years ago • 1 comments

I followed the instructions in README to train VoteNet on SUNRGBD data. All settings keep the default, but batch_size changed to 128(8GPUs). After 180 epoches, I got mAP results like this: 0.5466@IoU=0.25, 0.2811@IoU=0.5, which is much lower than that in paper(57 [email protected] and 32 [email protected]). If I use the pre-trained model, I can reproduce the results as paper noted, SO data is right.

Is batch_size matters so much? If so, how do I change the learning rate along with batch_size? Any idea is welcome! thanks!

StarsMyDestination avatar Dec 25 '19 09:12 StarsMyDestination

Have you solved this problem? How about using the default batch size? @StarsMyDestination

WhatAShot avatar May 22 '20 16:05 WhatAShot