code for pretraining on ImageNet

[adizhol-str]

Hello, thank you for your work and open-source contribution.
It is not clear to me from the paper how you pretrain on imagenet classification.
Specifically, what do you do with the "I" branch?

Can you share the code for this stage?

Thank you!