From decf74cbdf7a82ee5ecf47b31887f6ed3cdf09c6 Mon Sep 17 00:00:00 2001
From: Wendi Li <wendili.academic@qq.com>
Date: Mon, 1 Feb 2021 16:40:16 +0000
Subject: [PATCH] Update pytorch_gru.py

---
 qlib/contrib/model/pytorch_gru.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/qlib/contrib/model/pytorch_gru.py b/qlib/contrib/model/pytorch_gru.py
index 22c63c539..db8257093 100755
--- a/qlib/contrib/model/pytorch_gru.py
+++ b/qlib/contrib/model/pytorch_gru.py
@@ -56,7 +56,7 @@ class GRU(Model):
         early_stop=20,
         loss="mse",
         optimizer="adam",
-        GPU="0",
+        GPU=0,
         seed=None,
         **kwargs
     ):