Code

Rating   Name Size Complexity
A vit_model 278 38
A model 216 22
A _init_ 5 0
A train 212 12
A swin_transformer_model 211 20