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

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

diff --git a/qlib/contrib/model/pytorch_gats.py b/qlib/contrib/model/pytorch_gats.py
index c22e48204..9e5aa3e28 100644
--- a/qlib/contrib/model/pytorch_gats.py
+++ b/qlib/contrib/model/pytorch_gats.py
@@ -61,7 +61,7 @@ class GATs(Model):
         with_pretrain=True,
         model_path=None,
         optimizer="adam",
-        GPU="0",
+        GPU=0,
         seed=None,
         **kwargs
     ):