echo/pytorch-handbook
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

把1.3.5中的In[6]代码段对齐

Browse Source
This commit is contained in:
古代中二龙 2019-11-04 09:21:44 +08:00 committed by GitHub
parent c900b052c0
commit 860c8ad18f
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
chapter1/5_data_parallel_tutorial.ipynb
View File

@ -206,9 +206,9 @@
"source": [
"model = Model(input_size, output_size)\n",
"if torch.cuda.device_count() > 1:\n",
" print(\"Let's use\", torch.cuda.device_count(), \"GPUs!\")\n",
" # dim = 0 [30, xxx] -> [10, ...], [10, ...], [10, ...] on 3 GPUs\n",
" model = nn.DataParallel(model)\n",
" print(\"Let's use\", torch.cuda.device_count(), \"GPUs!\")\n",
" # dim = 0 [30, xxx] -> [10, ...], [10, ...], [10, ...] on 3 GPUs\n",
" model = nn.DataParallel(model)\n",
"\n",
"model.to(device)"
]