From ddd68fc761d3a7a4636de583dce9cb2cb1b00b20 Mon Sep 17 00:00:00 2001 From: Wendi Li Date: Mon, 1 Feb 2021 16:36:16 +0000 Subject: [PATCH] Update pytorch_alstm.py --- qlib/contrib/model/pytorch_alstm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_alstm.py b/qlib/contrib/model/pytorch_alstm.py index d85fef628..f2cfbdc36 100644 --- a/qlib/contrib/model/pytorch_alstm.py +++ b/qlib/contrib/model/pytorch_alstm.py @@ -56,7 +56,7 @@ class ALSTM(Model): early_stop=20, loss="mse", optimizer="adam", - GPU="0", + GPU=0, seed=None, **kwargs ):