Skip to content

Commit 6c59b6e

Browse files
laurenyuyuanzhua
authored andcommitted
Update MXNet ONNX Super-resolution notebook to use MXNet 1.3 (aws#763)
1 parent 868cbfa commit 6c59b6e

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

sagemaker-python-sdk/mxnet_onnx_superresolution/mxnet_onnx.ipynb

+4-3
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
"cell_type": "markdown",
55
"metadata": {},
66
"source": [
7-
"# Importing and hosting an ONNX model with MXNet 1.2.1\n",
7+
"# Importing and hosting an ONNX model with MXNet\n",
88
"\n",
99
"The [Open Neural Network Exchange](https://onnx.ai/) (ONNX) is an open format for representing deep learning models with its extensible computation graph model and definitions of built-in operators and standard data types.\n",
1010
"\n",
@@ -123,7 +123,7 @@
123123
"* `model_data`: the S3 location of the model data\n",
124124
"* `entry_point`: the script for model hosting that we looked at above\n",
125125
"* `role`: the IAM role used\n",
126-
"* `framework_version`: the MXNet version in use, in this case '1.2.1'\n",
126+
"* `framework_version`: the MXNet version in use\n",
127127
"\n",
128128
"You can read more about creating an `MXNetModel` object in the [SageMaker Python SDK API docs](https://sagemaker.readthedocs.io/en/latest/sagemaker.mxnet.html#mxnet-model)."
129129
]
@@ -139,7 +139,8 @@
139139
"mxnet_model = MXNetModel(model_data=model_data,\n",
140140
" entry_point='super_resolution.py',\n",
141141
" role=role,\n",
142-
" framework_version='1.2.1')"
142+
" py_version='py3',\n",
143+
" framework_version='1.3.0')"
143144
]
144145
},
145146
{

0 commit comments

Comments
 (0)