2021年7月24日
pytorch查看全连接层的权重和梯度
摘要： 比如，建了一个两层全连接层的神经网络： class LinearClassifier_2layer(nn.Module): def __init__(self, last_layer_dim=None, n_label=None): super(LinearClassifier_2layer, se 阅读全文
posted @ 2021-07-24 17:14 Picassooo 阅读(1996) 评论(0) 推荐(0) 编辑