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

Fix GPU

Browse Source
This commit is contained in:
Jactus
2020-12-02 12:23:42 +08:00
parent a2603fe27a
commit d109d3d44e
6 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/benchmarks/SFM/workflow_config_sfm.yaml
View File

@@ -57,7 +57,7 @@ task:
eval_steps: 5 eval_steps: 5
loss: mse loss: mse
optimizer: adam optimizer: adam
GPU: 1 GPU: 0
dataset: dataset:
class: DatasetH class: DatasetH
module_path: qlib.data.dataset module_path: qlib.data.dataset

2
qlib/contrib/model/pytorch_alstm.py
View File

@@ -135,7 +135,7 @@ class ALSTM(Model):
self.ALSTM_model.cuda() self.ALSTM_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"] = str(self.visible_GPU)
def mse(self, pred, label): def mse(self, pred, label):
loss = (pred - label) ** 2 loss = (pred - label) ** 2

2
qlib/contrib/model/pytorch_gats.py
View File

@@ -147,7 +147,7 @@ class GATs(Model):
self.GAT_model.cuda() self.GAT_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"] = str(self.visible_GPU)
def mse(self, pred, label): def mse(self, pred, label):
loss = (pred - label) ** 2 loss = (pred - label) ** 2

2
qlib/contrib/model/pytorch_gru.py
View File

@@ -135,7 +135,7 @@ class GRU(Model):
self.gru_model.cuda() self.gru_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"] = str(self.visible_GPU)
def mse(self, pred, label): def mse(self, pred, label):
loss = (pred - label) ** 2 loss = (pred - label) ** 2

2
qlib/contrib/model/pytorch_lstm.py
View File

@@ -135,7 +135,7 @@ class LSTM(Model):
self.lstm_model.cuda() self.lstm_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"] = str(self.visible_GPU)
def mse(self, pred, label): def mse(self, pred, label):
loss = (pred - label) ** 2 loss = (pred - label) ** 2

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

@@ -151,7 +151,7 @@ class DNNModelPytorch(Model):
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"] = str(self.visible_GPU)
def fit( def fit(
self, self,