Skip to content

Commit 4fef3b8

Browse files
committed
IR -> Mxnet refactor tested.
1 parent 6ff06ed commit 4fef3b8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

mmdnn/conversion/examples/mxnet/imagenet_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class TestMXNet(TestKit):
1616
def __init__(self):
1717
super(TestMXNet, self).__init__()
1818

19-
self.truth['tensorflow']['resnet'] = [(22, 13.58994), (147, 8.9273033), (90, 5.7173896), (24, 5.709712), (88, 4.7731524)]
19+
self.truth['tensorflow']['resnet'] = [(22, 11.8427), (147, 8.4613848), (24, 6.449892), (88, 4.2686858), (145, 4.1777167)]
2020
self.truth['tensorflow']['inception_v3'] = [(22, 9.6691055), (24, 4.3524752), (25, 3.5957956), (132, 3.5657482), (23, 3.3462858)]
2121
self.truth['keras']['inception_v3'] = [(21, 0.93430501), (23, 0.0028834261), (131, 0.0014781745), (24, 0.0014518937), (22, 0.0014435325)]
2222

mmdnn/conversion/mxnet/mxnet_emitter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -390,7 +390,7 @@ def _emit_convolution(self, IR_node, pattern):
390390
num_group = num_filter
391391
pattern = "Convolution"
392392
else:
393-
num_group = IR_node.IR_layer.attr["group"].i
393+
num_group = IR_node.get_attr('group', 1)
394394

395395
# layout = IR_node.IR_layer.attr["data_format"].s
396396
if dim == 1:

0 commit comments

Comments
 (0)