From 35840606a8721b031ad5ab63239a859a29782f98 Mon Sep 17 00:00:00 2001 From: Ying-Tao Luo Date: Fri, 16 Jul 2021 15:05:32 +0800 Subject: [PATCH] Update pytorch_localformer.py --- qlib/contrib/model/pytorch_localformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qlib/contrib/model/pytorch_localformer.py b/qlib/contrib/model/pytorch_localformer.py index bdf77b5be..683a9bd4f 100644 --- a/qlib/contrib/model/pytorch_localformer.py +++ b/qlib/contrib/model/pytorch_localformer.py @@ -42,7 +42,7 @@ class LocalformerModel(Model): optimizer="adam", reg=1e-3, n_jobs=10, - GPU=2, + GPU=0, seed=None, **kwargs ):