Skip to content

Commit 7cb4b3f

Browse files
author
Rui Wang Napieralski
committed
documentation: update PyTorch version in doc
1 parent 486cf44 commit 7cb4b3f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

doc/frameworks/pytorch/using_pytorch.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ directories ('train' and 'test').
154154
pytorch_estimator = PyTorch('pytorch-train.py',
155155
instance_type='ml.p3.2xlarge',
156156
instance_count=1,
157-
framework_version='1.5.0',
157+
framework_version='1.8.0',
158158
py_version='py3',
159159
hyperparameters = {'epochs': 20, 'batch-size': 64, 'learning-rate': 0.1})
160160
pytorch_estimator.fit({'train': 's3://my-data-bucket/path/to/my/training/data',
@@ -248,7 +248,7 @@ operation.
248248
pytorch_estimator = PyTorch(entry_point='train_and_deploy.py',
249249
instance_type='ml.p3.2xlarge',
250250
instance_count=1,
251-
framework_version='1.5.0',
251+
framework_version='1.8.0',
252252
py_version='py3')
253253
pytorch_estimator.fit('s3://my_bucket/my_training_data/')
254254

0 commit comments

Comments
 (0)