From 4c6c4f68c79e275018a6f0974c638c3f7ca90071 Mon Sep 17 00:00:00 2001 From: yh_cc Date: Sun, 5 May 2019 18:55:13 +0800 Subject: [PATCH] small change --- fastNLP/core/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastNLP/core/trainer.py b/fastNLP/core/trainer.py index b6c282b4..7ace58d9 100644 --- a/fastNLP/core/trainer.py +++ b/fastNLP/core/trainer.py @@ -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。