from datasets import load_dataset backbones_on_in1k_v1 = load_dataset("george-chou/vi_backbones", split="IMAGENET1K_V1") backbones_on_in1k_v2 = load_dataset("george-chou/vi_backbones", split="IMAGENET1K_V2") for weights in backbones_on_in1k_v1: print(weights) for weights in backbones_on_in1k_v2: print(weights)
git clone git@hf.co:datasets/george-chou/vi_backbones
https://pytorch.org/vision/main/_modules