saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-04 23:59:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor fixes in BERT Inference notebook (#5637)

Browse source 
Add missing commas to the code example.
This commit is contained in:
Derek Murray 2020-11-02 09:49:23 -08:00 committed by GitHub
parent 1cca903680
commit ff538b8d3a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/python/tools/transformers/notebooks/Inference_Bert_with_OnnxRuntime_on_AzureML.ipynb
View file

@ -115,9 +115,9 @@
"estimator = PyTorch(source_directory=project_roots, \n",
" script_params={'--output-dir': './outputs'},\n",
" compute_target=gpu_compute_target,\n",
" use_docker=True\n",
" custom_docker_image=image_name\n",
" script_params = {...}\n",
" use_docker=True,\n",
" custom_docker_image=image_name,\n",
" script_params = {...},\n",
" entry_script='run_squad_azureml.py', # change here\n",
" node_count=1,\n",
" process_count_per_node=4,\n",