model checkpoint meanings?

[dorsa-zeinali]

Hi,
Thank you for your work. I was wondering, what do the different model checkpoints mean? e.g. vit_h vs vit_l or vit_b

[pypae]

There is a paper accompanying the repository. The models are the same except for neural network size, B stands for "base" and is the smallest, L is "large" and H is "huge". The paper reports that the performance difference between L and H isn't much and I would recommend L if your machine supports it. However, B is lighter and not far behind in performance.

See #273 (comment)