Skip to content

Commit 860c8ad

Browse files
把1.3.5中的In[6]代码段对齐
1 parent c900b05 commit 860c8ad

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

chapter1/5_data_parallel_tutorial.ipynb

+3-3
Original file line numberDiff line numberDiff line change
@@ -206,9 +206,9 @@
206206
"source": [
207207
"model = Model(input_size, output_size)\n",
208208
"if torch.cuda.device_count() > 1:\n",
209-
" print(\"Let's use\", torch.cuda.device_count(), \"GPUs!\")\n",
210-
" # dim = 0 [30, xxx] -> [10, ...], [10, ...], [10, ...] on 3 GPUs\n",
211-
" model = nn.DataParallel(model)\n",
209+
" print(\"Let's use\", torch.cuda.device_count(), \"GPUs!\")\n",
210+
" # dim = 0 [30, xxx] -> [10, ...], [10, ...], [10, ...] on 3 GPUs\n",
211+
" model = nn.DataParallel(model)\n",
212212
"\n",
213213
"model.to(device)"
214214
]

0 commit comments

Comments
 (0)