pytorch查看网络层参数和梯度

        for name, parms in net.named_parameters():  
            print('-->name:', name)
            print('-->para:', parms)
            print('-->grad_requirs:',parms.requires_grad)
            print('-->grad_value:',parms.grad)
            print("===")
posted @ 2022-01-24 11:05  祥瑞哈哈哈  阅读(823)  评论(0)    收藏  举报