Question about the class token

[Andy1621]

Good job!
My question is that why to use different class tokens for each stage but only the final class token is used for classification?

CvT/lib/models/cls_cvt.py

Line 607 in 34d1af9

x, cls_tokens = getattr(self, f'stage{i}')(x)