graph-rcnn.pytorch icon indicating copy to clipboard operation
graph-rcnn.pytorch copied to clipboard

Train "sg_grcnn" with batch_size=1

Open jgyy4775 opened this issue 5 years ago • 2 comments

I trained this model with the command "python main.py --config-file configs/sgg_res101_step.yaml --algorithm sg_grcnn". SOLVER: BASE_LR: 5e-3 MAX_ITER: 8000 STEPS: (8000,12000) batch_size : 1

But the result is very strange... image Are the steps and batch too small? I wonder why the performance is low.

If not for batch and step did I make a mistake?

jgyy4775 avatar Aug 29 '19 13:08 jgyy4775

hello,do you know why?,i have the same question.

August66 avatar Apr 05 '20 03:04 August66

I have the same question too.Have you solved the problem yet?

diaoran avatar Apr 11 '22 08:04 diaoran