liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-01 18:11:18 +08:00
Code Activity

update US config to us

Browse Source
This commit is contained in:
bxdd
2020-10-12 14:46:31 +08:00
committed by you-n-g
parent 916d9d363e
commit eda0c81bfb
2 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
qlib/config.py
View File

@@ -60,7 +60,7 @@ class Config:
# REGION CONST
REG_CN = "cn"
REG_US = "US"
REG_US = "us"
_default_config = {
# data provider config

3
qlib/contrib/model/pytorch_nn.py
View File

@@ -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