Skip to content

Commit 81b4033

Browse files
authored
Fix mrcnn tf2 notebook (#130)
* update file pattern * drop empty line
1 parent 1d00256 commit 81b4033

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

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

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -217,15 +217,14 @@
217217
" \"total_steps\": 500,\n",
218218
" \"train_batch_size\": 4,\n",
219219
" \"eval_batch_size\": 8,\n",
220-
" \"training_file_pattern\": '/'.join([SM_DATA_ROOT, 'train/train*.tfrecord']), \n",
221-
" \"validation_file_pattern\": '/'.join([SM_DATA_ROOT, 'val/val*.tfrecord']), \n",
220+
" \"training_file_pattern\": '/'.join([SM_DATA_ROOT, 'train']), \n",
221+
" \"validation_file_pattern\": '/'.join([SM_DATA_ROOT, 'val']), \n",
222222
" \"val_json_file\": '/'.join([SM_DATA_ROOT, 'annotations/instances_val2017.json']), \n",
223223
" \"amp\": '',\n",
224224
" \"use_batched_nms\": '',\n",
225225
" \"xla\": '',\n",
226226
" \"nouse_custom_box_proposals_op\": '',\n",
227227
" \"seed\": 987\n",
228-
" \n",
229228
" }"
230229
]
231230
},
@@ -324,4 +323,4 @@
324323
},
325324
"nbformat": 4,
326325
"nbformat_minor": 4
327-
}
326+
}

0 commit comments

Comments
 (0)