When you asked me to send you a pull request, I checked the code again.
And found that there is a little problem in printing infos. And I annotated it.
As for the layer_num and hidden_size, there is a missing assignment in the class.
So the self.num_layers is missing. And I've already added it.
Dear Professor,
When you asked me to send you a pull request, I checked the code again. And found that there is a little problem in printing infos. And I annotated it.
As for the layer_num and hidden_size, there is a missing assignment in the class. So the
self.num_layers
is missing. And I've already added it.Best, Zilin