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

Add return for use_gpu..

Browse Source
This commit is contained in:
D-X-Y
2021-03-11 19:28:00 -08:00
parent 593553f573
commit 1d435248e2
11 changed files with 31 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -99,8 +99,8 @@ class DNNModelPytorch(Model):
"\nloss_type : {}"
"\neval_steps : {}"
"\nseed : {}"
"\nvisible_GPU : {}"
"\nuse_gpu : {}"
"\ndevice : {}"
"\nuse_GPU : {}"
"\nweight_decay : {}".format(
layers,
lr,
@@ -114,7 +114,7 @@ class DNNModelPytorch(Model):
loss,
eval_steps,
seed,
GPU,
self.device,
self.use_gpu,
weight_decay,
)
@@ -158,7 +158,7 @@ class DNNModelPytorch(Model):
@property
def use_gpu(self):
self.device != torch.device("cpu")
return self.device != torch.device("cpu")
def fit(
self,