Skip to content

Commit c9fa95f

Browse files
authored
Merge pull request #251 from zhreshold/add-export-cifar-models
add cifar exported models
2 parents ef0a0af + 221ce5b commit c9fa95f

15 files changed

+48693
-0
lines changed
6.68 MB
Binary file not shown.

gluoncv/models/cifar_resnet110_v1-symbol.json

+13,746
Large diffs are not rendered by default.
6.68 MB
Binary file not shown.

gluoncv/models/cifar_resnet110_v2-symbol.json

+13,685
Large diffs are not rendered by default.
1.05 MB
Binary file not shown.

gluoncv/models/cifar_resnet20_v1-symbol.json

+2,631
Large diffs are not rendered by default.
1.05 MB
Binary file not shown.

gluoncv/models/cifar_resnet20_v2-symbol.json

+2,570
Large diffs are not rendered by default.
3.31 MB
Binary file not shown.

gluoncv/models/cifar_resnet56_v1-symbol.json

+7,077
Large diffs are not rendered by default.
3.3 MB
Binary file not shown.

gluoncv/models/cifar_resnet56_v2-symbol.json

+7,016
Large diffs are not rendered by default.
65.3 MB
Binary file not shown.

gluoncv/models/cifar_wideresnet16_10-symbol.json

+1,939
Large diffs are not rendered by default.

gluoncv/models/export.py

+29
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
import gluoncv
2+
import mxnet as mx
3+
4+
5+
models = [
6+
'cifar_resnet20_v1',
7+
'cifar_resnet56_v1',
8+
'cifar_resnet110_v1',
9+
'cifar_resnet20_v2',
10+
'cifar_resnet56_v2',
11+
'cifar_resnet110_v2',
12+
'cifar_wideresnet16_10',
13+
#'cifar_wideresnet28_10',
14+
#'cifar_wideresnet40_8',
15+
#'cifar_resnext29_16x64d'
16+
]
17+
18+
def export(model_name):
19+
h, w = 32, 32
20+
net = gluoncv.model_zoo.get_model(model_name, pretrained=True)
21+
net.hybridize()
22+
net.forward(mx.nd.zeros((1, 3, h, w)))
23+
net.export(model_name)
24+
25+
26+
if __name__ == '__main__':
27+
for model in models:
28+
print(model)
29+
export(model)

0 commit comments

Comments
 (0)