Skip to content

Commit 18d362e

Browse files
author
Chuyang Deng
committed
undo pytorch default version change
1 parent a6378fd commit 18d362e

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

README.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ PyTorch SageMaker Estimators
205205

206206
With PyTorch SageMaker Estimators, you can train and host PyTorch models on Amazon SageMaker.
207207

208-
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``.
208+
Supported versions of PyTorch: ``0.4.0``, ``1.0.0``, ``1.1.0``.
209209

210210
We recommend that you use the latest supported version, because that's where we focus most of our development efforts.
211211

src/sagemaker/pytorch/defaults.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
# language governing permissions and limitations under the License.
1313
from __future__ import absolute_import
1414

15-
PYTORCH_VERSION = '1.1.0'
15+
PYTORCH_VERSION = '0.4'
1616
"""Default PyTorch version for when the framework version is not specified.
1717
This is no longer updated so as to not break existing workflows.
1818
"""

0 commit comments

Comments
 (0)