我需要对CNN模型的每一个历元分别取权值。我有100个历元，所以我需要看看权值是如何变化的。比如，第一阶段：{权重}，第2阶段：{weights}等等。我正在做“model.Cnn1.weight”，但是它可能打印了该层的所有权重。请给予我任何建议。谢谢
jckbn6z71#
可以使用以下代码打印权重值。apply方法递归地将函数应用于模型中的所有层。和parameter_print函数检查它是线性层还是卷积层并打印权重值。
def parameter_print(m): if isinstance(m, nn.Conv2d) or isinstance(m, nn.Linear): print(m.weight) model.apply(parameter_print)
1条答案
按热度按时间jckbn6z71#
可以使用以下代码打印权重值。
apply方法递归地将函数应用于模型中的所有层。
和parameter_print函数检查它是线性层还是卷积层并打印权重值。