Skip to content

Commit bd122a1

Browse files
authored
SM SDK v2 name changes (#121)
* update parameters for PT MaskRNN for SM SDK v2 * Update instance type & instance count for PT BERT * Update TF2 BERT * Update tensorflow2_smdataparallel_maskrcnn_demo.ipynb
1 parent ae9e612 commit bd122a1

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

training/distributed_training/pytorch/data_parallel/bert/pytorch_smdataparallel_bert_demo.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -240,8 +240,8 @@
240240
" role=role,\n",
241241
" image_uri=docker_image,\n",
242242
" framework_version='1.6.0',\n",
243-
" train_instance_count=instance_count,\n",
244-
" train_instance_type=instance_type,\n",
243+
" instance_count=instance_count,\n",
244+
" instance_type=instance_type,\n",
245245
" sagemaker_session=sagemaker_session,\n",
246246
" subnets=subnets,\n",
247247
" security_group_ids=security_group_ids,\n",

training/distributed_training/pytorch/data_parallel/maskrcnn/pytorch_smdataparallel_maskrcnn_demo.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -210,8 +210,8 @@
210210
" role=role,\n",
211211
" image_uri=docker_image,\n",
212212
" source_dir='.',\n",
213-
" train_instance_count=instance_count,\n",
214-
" train_instance_type=instance_type,\n",
213+
" instance_count=instance_count,\n",
214+
" instance_type=instance_type,\n",
215215
" sagemaker_session=sagemaker_session,\n",
216216
" hyperparameters=hyperparameters, \n",
217217
" subnets=subnets, \n",

training/distributed_training/tensorflow/data_parallel/bert/tensorflow2_smdataparallel_bert_demo.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -245,8 +245,8 @@
245245
" image_uri=docker_image,\n",
246246
" source_dir='deep-learning-models/models/nlp',\n",
247247
" framework_version='2.3.1',\n",
248-
" train_instance_count=instance_count,\n",
249-
" train_instance_type=instance_type,\n",
248+
" instance_count=instance_count,\n",
249+
" instance_type=instance_type,\n",
250250
" sagemaker_session=sagemaker_session,\n",
251251
" subnets=subnets,\n",
252252
" hyperparameters=hyperparameters,\n",

training/distributed_training/tensorflow/data_parallel/maskrcnn/tensorflow2_smdataparallel_maskrcnn_demo.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -240,8 +240,8 @@
240240
" image_uri=docker_image,\n",
241241
" source_dir='.',\n",
242242
" framework_version='2.3.1',\n",
243-
" train_instance_count=instance_count,\n",
244-
" train_instance_type=instance_type,\n",
243+
" instance_count=instance_count,\n",
244+
" instance_type=instance_type,\n",
245245
" sagemaker_session=sagemaker_session,\n",
246246
" subnets=subnets,\n",
247247
" hyperparameters=hyperparameters,\n",

0 commit comments

Comments
 (0)