File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -154,7 +154,7 @@ directories ('train' and 'test').
154
154
pytorch_estimator = PyTorch(' pytorch-train.py' ,
155
155
instance_type = ' ml.p3.2xlarge' ,
156
156
instance_count = 1 ,
157
- framework_version = ' 1.5 .0' ,
157
+ framework_version = ' 1.8 .0' ,
158
158
py_version = ' py3' ,
159
159
hyperparameters = {' epochs' : 20 , ' batch-size' : 64 , ' learning-rate' : 0.1 })
160
160
pytorch_estimator.fit({' train' : ' s3://my-data-bucket/path/to/my/training/data' ,
@@ -248,7 +248,7 @@ operation.
248
248
pytorch_estimator = PyTorch(entry_point = ' train_and_deploy.py' ,
249
249
instance_type = ' ml.p3.2xlarge' ,
250
250
instance_count = 1 ,
251
- framework_version = ' 1.5 .0' ,
251
+ framework_version = ' 1.8 .0' ,
252
252
py_version = ' py3' )
253
253
pytorch_estimator.fit(' s3://my_bucket/my_training_data/' )
254
254
You can’t perform that action at this time.
0 commit comments