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
small change
tags/v0.4.10
yh_cc
6 years ago
parent
ded4228f93
commit
4c6c4f68c7
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/trainer.py
+ 1
- 1
fastNLP/core/trainer.py
View File
@@ -366,7 +366,7 @@ class Trainer(object):
2. torch.device:将模型装载到torch.device上。
3. int: 将使用device
_id为该值
的gpu进行训练
3. int: 将使用
该
device的gpu进行训练
4. list(int):如果多于1个device,将使用torch.nn.DataParallel包裹model, 并使用传入的device。
Write
Preview
Loading…
Cancel
Save