Image Classification
timm
PDE
ConvNet
liuyao commited on
Commit
8c20fb4
1 Parent(s): 3f1b078

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -65,7 +65,7 @@ import torch, timm
65
  from qlnet import QLNet
66
 
67
  model = QLNet()
68
- model.load_state_dict(torch.load('qlnet-50-v0.pth.tar')['state_dict'])
69
  model.eval()
70
  ```
71
 
@@ -87,4 +87,4 @@ python3 train.py ../datasets/imagenet/ --model resnet50 --num-classes 1000 --lr
87
 
88
  ### Results
89
 
90
- qlnet-50-v0: acc=78.608
 
65
  from qlnet import QLNet
66
 
67
  model = QLNet()
68
+ model.load_state_dict(torch.load('qlnet-10m.pth.tar'))
69
  model.eval()
70
  ```
71
 
 
87
 
88
  ### Results
89
 
90
+ qlnet-10m: acc=78.608