We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 3886022 + 56eaacd commit 0387eafCopy full SHA for 0387eaf
qlib/contrib/model/pytorch_gru_ts.py
@@ -126,8 +126,8 @@ def __init__(
126
num_layers=self.num_layers,
127
dropout=self.dropout,
128
)
129
- self.logger.info("model:\n{:}".format(self.gru_model))
130
- self.logger.info("model size: {:.4f} MB".format(count_parameters(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)))
131
132
if optimizer.lower() == "adam":
133
self.train_optimizer = optim.Adam(self.GRU_model.parameters(), lr=self.lr)
0 commit comments