result problem

Question

result problem

Opened this issue 7 months ago · 4 comments

It is possible to ask whether the results of ViT-B on the SBD training set were trained with 55 epochs or 230 brother epochs.
I ran 55 epochs with the following results

Answer 1 · 2024-05-16T07:02:29.000Z

@caibajin Hi, all ViT-B models were trained with 55 epochs. It seems your results are much worse than ours. Did you use the same training settings as specified here: https://github.com/uncbiag/SimpleClick/blob/v1.0/models/iter_mask/plainvit_base448_sbd_itermask.py? Meanwhile, our batch size is 140 for ViT-B models. Let me know your training settings.

Answer 2 · 2024-05-16T07:17:52.000Z

device：4090
batch-size：24

Answer 3 · 2024-05-16T10:33:26.000Z

Hi, I'd like to ask if it's related to batch-size?

Answer 4 · 2024-05-16T14:09:54.000Z

@caibajin It's less likely that using a batch size of 24 instead of 140 would greatly impact the performance, though you can have a try if that's the only difference between your training settings and mine. BTW, I used 4 A6000, but 4090 should be able to reproduce the results.