From 213f809148a55933506368a0c9592ce9a6d33d77 Mon Sep 17 00:00:00 2001 From: Wendi Li Date: Mon, 1 Feb 2021 16:37:36 +0000 Subject: [PATCH] Update pytorch_alstm_ts.py --- qlib/contrib/model/pytorch_alstm_ts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_alstm_ts.py b/qlib/contrib/model/pytorch_alstm_ts.py index fabdec831..d2a5db8f1 100644 --- a/qlib/contrib/model/pytorch_alstm_ts.py +++ b/qlib/contrib/model/pytorch_alstm_ts.py @@ -58,7 +58,7 @@ class ALSTM(Model): loss="mse", optimizer="adam", n_jobs=10, - GPU="0", + GPU=0, seed=None, **kwargs ):