From eda0c81bfbc7d375e93832b884d1d43fce090b09 Mon Sep 17 00:00:00 2001 From: bxdd Date: Mon, 12 Oct 2020 14:46:31 +0800 Subject: [PATCH] update US config to us --- qlib/config.py | 2 +- qlib/contrib/model/pytorch_nn.py | 3 --- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/qlib/config.py b/qlib/config.py index 4dea8fb43..ff01fe5e8 100644 --- a/qlib/config.py +++ b/qlib/config.py @@ -60,7 +60,7 @@ class Config: # REGION CONST REG_CN = "cn" -REG_US = "US" +REG_US = "us" _default_config = { # data provider config diff --git a/qlib/contrib/model/pytorch_nn.py b/qlib/contrib/model/pytorch_nn.py index b27664848..48b643bf8 100644 --- a/qlib/contrib/model/pytorch_nn.py +++ b/qlib/contrib/model/pytorch_nn.py @@ -136,11 +136,8 @@ class DNNModelPytorch(Model): ) self._fitted = False - - if self.use_gpu: self.dnn_model.cuda() - # set the visible GPU if self.visible_GPU: os.environ["CUDA_VISIBLE_DEVICES"] = self.visible_GPU