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

Merge pull request #373 from lewwang1995/main

Browse Source 
debug
This commit is contained in:
you-n-g
2021-03-26 11:59:20 +08:00
committed by GitHub
parent 3886022669 56eaacd931
commit 0387eaf7ab
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
qlib/contrib/model/pytorch_gru_ts.py
View File

@@ -126,8 +126,8 @@ class GRU(Model):
num_layers=self.num_layers, num_layers=self.num_layers,
dropout=self.dropout, dropout=self.dropout,
) )
self.logger.info("model:\n{:}".format(self.gru_model)) self.logger.info("model:\n{:}".format(self.GRU_model))
self.logger.info("model size: {:.4f} MB".format(count_parameters(self.gru_model))) self.logger.info("model size: {:.4f} MB".format(count_parameters(self.GRU_model)))
if optimizer.lower() == "adam": if optimizer.lower() == "adam":
self.train_optimizer = optim.Adam(self.GRU_model.parameters(), lr=self.lr) self.train_optimizer = optim.Adam(self.GRU_model.parameters(), lr=self.lr)