You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
@@ -767,7 +767,8 @@ This customized Python code must be named ``inference.py`` and is specified thro
767
767
768
768
model = TensorFlowModel(entry_point='inference.py',
769
769
model_data='s3://mybucket/model.tar.gz',
770
-
role='MySageMakerRole')
770
+
role='MySageMakerRole',
771
+
framework_version='x.x.x')
771
772
772
773
In the example above, ``inference.py`` is assumed to be a file inside ``model.tar.gz``. If you want to use a local file instead, you must add the ``source_dir`` argument. See the documentation on `TensorFlowModel <https://sagemaker.readthedocs.io/en/stable/frameworks/tensorflow/sagemaker.tensorflow.html#sagemaker.tensorflow.model.TensorFlowModel>`_.
773
774
@@ -923,7 +924,8 @@ processing. There are 2 ways to do this:
923
924
model = TensorFlowModel(entry_point='inference.py',
924
925
dependencies=['requirements.txt'],
925
926
model_data='s3://mybucket/model.tar.gz',
926
-
role='MySageMakerRole')
927
+
role='MySageMakerRole',
928
+
framework_version='x.x.x')
927
929
928
930
929
931
2. If you are working in a network-isolation situation or if you don't
@@ -941,7 +943,8 @@ processing. There are 2 ways to do this:
941
943
model = TensorFlowModel(entry_point='inference.py',
942
944
dependencies=['/path/to/folder/named/lib'],
943
945
model_data='s3://mybucket/model.tar.gz',
944
-
role='MySageMakerRole')
946
+
role='MySageMakerRole',
947
+
framework_version='x.x.x')
945
948
946
949
For more information, see: https://github.com/aws/sagemaker-tensorflow-serving-container#prepost-processing
0 commit comments