Skip to content

Commit

Permalink
Fix malformated code in jupyter notebook (#5512)
Browse files Browse the repository at this point in the history
Signed-off-by: Janusz Lisiecki <[email protected]>
  • Loading branch information
JanuszL authored Jun 10, 2024
1 parent e7526de commit 81ebc6a
Showing 1 changed file with 6 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -260,8 +260,12 @@
"def get_batch_test_dali(batch_size, pipe_type):\n",
" pipe_name, label_type, _ = pipe_type\n",
" pipes = [\n",
" pipe_name(batch_size=BATCH_SIZE, num_threads=2, device_id=device_id, n\n",
" um_gpus=N)\n",
" pipe_name(\n",
" batch_size=BATCH_SIZE,\n",
" num_threads=2,\n",
" device_id=device_id,\n",
" num_gpus=N,\n",
" )\n",
" for device_id in range(N)\n",
" ]\n",
"\n",
Expand Down

0 comments on commit 81ebc6a

Please sign in to comment.