This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
Register
Sign In
hummingbird
/
fastNLP
Not watched
Unwatch
Watch all
Watch but not notify
2
Star
0
Fork
0
Code
Releases
13
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
Browse Source
修复一个data_parallel情况下的gpu不从0开始导致的bug
tags/v0.6.0
yh_cc
5 years ago
parent
a6c2533bd3
commit
ed2309719c
1 changed files
with
1 additions
and
1 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
fastNLP/core/utils.py
+ 1
- 1
fastNLP/core/utils.py
View File
@@ -409,7 +409,7 @@ def _move_model_to_device(model, device):
if device is None:
if isinstance(model, torch.nn.DataParallel):
model.cuda()
model.cuda(
model.device_ids[0]
)
return model
else:
if not torch.cuda.is_available() and ((isinstance(device, str) and device!='cpu') or
Write
Preview
Loading…
Cancel
Save