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
qlib/contrib/model/pytorch_lstm.py
View File

@@ -135,7 +135,7 @@ class LSTM(Model):
self.lstm_model.cuda()
# set the 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):
loss = (pred - label) ** 2