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

Update devices

Browse Source
This commit is contained in:
Jactus
2020-12-02 13:28:32 +08:00
parent 7f385345bb
commit 745b93138d
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_alstm.py
View File

@@ -76,7 +76,7 @@ class ALSTM(Model):
self.early_stop = early_stop self.early_stop = early_stop
self.optimizer = optimizer.lower() self.optimizer = optimizer.lower()
self.loss = loss self.loss = loss
self.device = "cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu" self.device = torch.device("cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu")
self.use_gpu = torch.cuda.is_available() self.use_gpu = torch.cuda.is_available()
self.seed = seed self.seed = seed

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

@@ -83,7 +83,7 @@ class GATs(Model):
self.base_model = base_model self.base_model = base_model
self.with_pretrain = with_pretrain self.with_pretrain = with_pretrain
self.model_path = model_path self.model_path = model_path
self.device = "cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu" self.device = torch.device("cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu")
self.use_gpu = torch.cuda.is_available() self.use_gpu = torch.cuda.is_available()
self.seed = seed self.seed = seed

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

@@ -76,7 +76,7 @@ class GRU(Model):
self.early_stop = early_stop self.early_stop = early_stop
self.optimizer = optimizer.lower() self.optimizer = optimizer.lower()
self.loss = loss self.loss = loss
self.device = "cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu" self.device = torch.device("cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu")
self.use_gpu = torch.cuda.is_available() self.use_gpu = torch.cuda.is_available()
self.seed = seed self.seed = seed

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

@@ -76,7 +76,7 @@ class LSTM(Model):
self.early_stop = early_stop self.early_stop = early_stop
self.optimizer = optimizer.lower() self.optimizer = optimizer.lower()
self.loss = loss self.loss = loss
self.device = "cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu" self.device = torch.device("cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu")
self.use_gpu = torch.cuda.is_available() self.use_gpu = torch.cuda.is_available()
self.seed = seed self.seed = seed

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

@@ -79,7 +79,7 @@ class DNNModelPytorch(Model):
self.lr_decay_steps = lr_decay_steps self.lr_decay_steps = lr_decay_steps
self.optimizer = optimizer.lower() self.optimizer = optimizer.lower()
self.loss_type = loss self.loss_type = loss
self.device = "cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu" self.device = torch.device("cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu")
self.use_GPU = torch.cuda.is_available() self.use_GPU = torch.cuda.is_available()
self.seed = seed self.seed = seed

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

@@ -239,7 +239,7 @@ class SFM(Model):
self.eval_steps = eval_steps self.eval_steps = eval_steps
self.optimizer = optimizer.lower() self.optimizer = optimizer.lower()
self.loss = loss self.loss = loss
self.device = "cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu" self.device = torch.device("cuda:%d" % (GPU) if torch.cuda.is_available() else "cpu")
self.use_gpu = torch.cuda.is_available() self.use_gpu = torch.cuda.is_available()
self.seed = seed self.seed = seed