search

uncbiag / SimpleClick

SimpleClick: Interactive Image Segmentation with Simple Vision Transformers (ICCV 2023)
MIT License
209 stars 32 forks source link

result problem #34

Open caibajin opened 4 months ago

caibajin commented 4 months ago

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 屏幕截图 2024-05-16 000843

qinliuliuqin commented 4 months ago

@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.

caibajin commented 4 months ago

device：4090 batch-size：24

caibajin commented 4 months ago

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

qinliuliuqin commented 4 months ago

@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.