From e006ef40add859597ca94bb3a27064b33b27576d Mon Sep 17 00:00:00 2001 From: Ying-Tao Luo Date: Sun, 18 Jul 2021 22:37:41 +0800 Subject: [PATCH] Update pytorch_localformer_ts.py --- qlib/contrib/model/pytorch_localformer_ts.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/qlib/contrib/model/pytorch_localformer_ts.py b/qlib/contrib/model/pytorch_localformer_ts.py index b4999bc9c..683a9bd4f 100644 --- a/qlib/contrib/model/pytorch_localformer_ts.py +++ b/qlib/contrib/model/pytorch_localformer_ts.py @@ -42,7 +42,7 @@ class LocalformerModel(Model): optimizer="adam", reg=1e-3, n_jobs=10, - GPU=2, + GPU=0, seed=None, **kwargs ): @@ -153,9 +153,6 @@ class LocalformerModel(Model): dl_train = dataset.prepare("train", col_set=["feature", "label"], data_key=DataHandlerLP.DK_L) dl_valid = dataset.prepare("valid", col_set=["feature", "label"], data_key=DataHandlerLP.DK_L) - import pdb - - pdb.set_trace() dl_train.config(fillna_type="ffill+bfill") # process nan brought by dataloader dl_valid.config(fillna_type="ffill+bfill") # process nan brought by dataloader