Skip to content

Commit 0387eaf

Browse files
authored
Merge pull request microsoft#373 from lewwang1995/main
debug
2 parents 3886022 + 56eaacd commit 0387eaf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

qlib/contrib/model/pytorch_gru_ts.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -126,8 +126,8 @@ def __init__(
126126
num_layers=self.num_layers,
127127
dropout=self.dropout,
128128
)
129-
self.logger.info("model:\n{:}".format(self.gru_model))
130-
self.logger.info("model size: {:.4f} MB".format(count_parameters(self.gru_model)))
129+
self.logger.info("model:\n{:}".format(self.GRU_model))
130+
self.logger.info("model size: {:.4f} MB".format(count_parameters(self.GRU_model)))
131131

132132
if optimizer.lower() == "adam":
133133
self.train_optimizer = optim.Adam(self.GRU_model.parameters(), lr=self.lr)

0 commit comments

Comments
 (0)