From 1c8174f59cc1e51f8b9a1f28f758adbd3bad28e7 Mon Sep 17 00:00:00 2001
From: Cola <43774355+Colanim@users.noreply.github.com>
Date: Wed, 23 Oct 2019 16:25:32 +0900
Subject: [PATCH] device id -1 as int (baseline)

For baseline function, device ID is given as a constant (-1, because baseline don't need GPU). This constant was a `str`, giving TypeError. Modified to int.
---
 src/train_abstractive.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/train_abstractive.py b/src/train_abstractive.py
index 545efde3..1ea3e3f9 100644
--- a/src/train_abstractive.py
+++ b/src/train_abstractive.py
@@ -258,7 +258,7 @@ def baseline(args, cal_lead=False, cal_oracle=False):
                                        args.batch_size, 'cpu',
                                        shuffle=False, is_test=True)
 
-    trainer = build_trainer(args, '-1', None, None, None)
+    trainer = build_trainer(args, -1, None, None, None)
     #
     if (cal_lead):
         trainer.test(test_iter, 0, cal_lead=True)