Image Classification
timm
PDE
ConvNet
liuyao commited on
Commit
74a083d
1 Parent(s): f2a94e5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -1
README.md CHANGED
@@ -45,7 +45,14 @@ Instead of the `bottleneck` block of ResNet50 which consists of 1x1, 3x3, 1x1 in
45
 
46
  Use the code below to get started with the model.
47
 
48
- [More Information Needed]
 
 
 
 
 
 
 
49
 
50
  ## Training Details
51
 
 
45
 
46
  Use the code below to get started with the model.
47
 
48
+ ```
49
+ import torch, timm
50
+ from qlnet import QLNet
51
+
52
+ model = QLNet()
53
+ model.load_state_dict(torch.load('qlnet-50-v0.pth.tar')['state_dict'])
54
+ model.eval()
55
+ ```
56
 
57
  ## Training Details
58