MuGeminorum commited on
Commit
e8ff109
1 Parent(s): 5295064
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -18,7 +18,7 @@ def get_backbone(ver, backbone_list):
18
  def model_info(m_ver):
19
  backbone_list = load_dataset(
20
  "monet-joe/cv_backbones",
21
- split="ImageNet1k_v1",
22
  trust_remote_code=True,
23
  )
24
  backbone = get_backbone(m_ver, backbone_list)
 
18
  def model_info(m_ver):
19
  backbone_list = load_dataset(
20
  "monet-joe/cv_backbones",
21
+ split="IMAGENET1K_V1",
22
  trust_remote_code=True,
23
  )
24
  backbone = get_backbone(m_ver, backbone_list)