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

Fix bug in MLP

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

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

@@ -158,7 +158,7 @@ class DNNModelPytorch(Model):
@property @property
def use_gpu(self): def use_gpu(self):
self.device == torch.device("cpu") self.device != torch.device("cpu")
def fit( def fit(
self, self,
@@ -222,7 +222,8 @@ class DNNModelPytorch(Model):
# validation # validation
train_loss += loss.val train_loss += loss.val
if step and step % self.eval_steps == 0: # for evert `eval_steps` steps or at the last steps, we will evaluate the model.
if step % self.eval_steps == 0 or step + 1 == self.max_steps:
stop_steps += 1 stop_steps += 1
train_loss /= self.eval_steps train_loss /= self.eval_steps
@@ -255,7 +256,7 @@ class DNNModelPytorch(Model):
# update learning rate # update learning rate
self.scheduler.step(cur_loss_val) self.scheduler.step(cur_loss_val)
# restore the optimal parameters after training ?? # restore the optimal parameters after training
self.dnn_model.load_state_dict(torch.load(save_path)) self.dnn_model.load_state_dict(torch.load(save_path))
if self.use_gpu: if self.use_gpu:
torch.cuda.empty_cache() torch.cuda.empty_cache()