xedwin23x commited on
Commit
9f7c0b9
·
verified ·
1 Parent(s): e84c2cb

Card: load with fgir_zoo.tgda

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -84,9 +84,9 @@ class count, accuracy, SHA-256 and size.
84
  import torch
85
  from PIL import Image
86
  from torchvision import transforms
87
- from fgir_zoo import fgirkd
88
 
89
- model = fgirkd.create_model('students/serial_203/cub_resnet18_vgg19_bn_cal_203')
90
  cfg = model.config
91
  tf = transforms.Compose([
92
  transforms.Resize((cfg.test_resize_size, cfg.test_resize_size),
 
84
  import torch
85
  from PIL import Image
86
  from torchvision import transforms
87
+ from fgir_zoo import tgda
88
 
89
+ model = tgda.create_model('students/serial_203/cub_resnet18_vgg19_bn_cal_203')
90
  cfg = model.config
91
  tf = transforms.Compose([
92
  transforms.Resize((cfg.test_resize_size, cfg.test_resize_size),