liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 23:36:54 +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 # REGION CONST
REG_CN = "cn" REG_CN = "cn"
REG_US = "US" REG_US = "us"
_default_config = { _default_config = {
# data provider config # data provider config

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

@@ -136,11 +136,8 @@ class DNNModelPytorch(Model):
) )
self._fitted = False self._fitted = False
if self.use_gpu: if self.use_gpu:
self.dnn_model.cuda() self.dnn_model.cuda()
# set the visible GPU # set the visible GPU
if self.visible_GPU: if self.visible_GPU:
os.environ["CUDA_VISIBLE_DEVICES"] = self.visible_GPU os.environ["CUDA_VISIBLE_DEVICES"] = self.visible_GPU