fix: resolve breaking changes of neo container, adding softmax_label to compile_model (#1635)

chuyang-deng · web-flow · commit 181f8d6d96f2 · 2020-10-20T10:01:50.000-07:00
diff --git a/sagemaker-python-sdk/mxnet_mnist/mxnet_mnist.ipynb b/sagemaker-python-sdk/mxnet_mnist/mxnet_mnist.ipynb
@@ -154,7 +154,7 @@
     "    output_path = '/'.join(mnist_estimator.output_path.split('/')[:-1])\n",
     "    neo_optimize = True\n",
     "    compiled_model = mnist_estimator.compile_model(target_instance_family='ml_m4', \n",
-    "                                                   input_shape={'data':[1, 784]},\n",
+    "                                                   input_shape={'data':[1, 784], 'softmax_label': [1]},\n",
     "                                                   role=role,\n",
     "                                                   output_path=output_path)"
    ]

Original file line number	Diff line number	Diff line change
`@@ -154,7 +154,7 @@`
`154`	`154`	`" output_path = '/'.join(mnist_estimator.output_path.split('/')[:-1])\n",`
`155`	`155`	`" neo_optimize = True\n",`
`156`	`156`	`" compiled_model = mnist_estimator.compile_model(target_instance_family='ml_m4', \n",`
`157`		`- " input_shape={'data':[1, 784]},\n",`
	`157`	`+ " input_shape={'data':[1, 784], 'softmax_label': [1]},\n",`
`158`	`158`	`" role=role,\n",`
`159`	`159`	`" output_path=output_path)"`
`160`	`160`	`]`