liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-11 23:06:58 +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

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