Hugging Face's logo

cis519
/
efficient-Net

Safetensors
Model card Files Community
DanioZhong commited on
Commit
0d1f37d
·
verified ·
1 Parent(s): badc963

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ from PIL import Image
37
  import os
38
  import numpy as np
39
 
40
- class EfficientNetGPSModel(nn.Module, Hubmixin):
41
  def __init__(self, eff_name="efficientnet_b0", num_outputs=2):
42
  super(EfficientNetGPSModel, self).__init__()
43
 
 
37
  import os
38
  import numpy as np
39
 
40
+ class EfficientNetGPSModel(nn.Module, PyTorchModelHubMixin):
41
  def __init__(self, eff_name="efficientnet_b0", num_outputs=2):
42
  super(EfficientNetGPSModel, self).__init__()
43