Browse Source

small change

tags/v0.4.10
yh_cc 6 years ago
parent
commit
4c6c4f68c7
1 changed files with 1 additions and 1 deletions
  1. +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。



Loading…
Cancel
Save