pytorch打印模型参数

目录

torchsummary打印

打印模型参数,两种方法结果好像是一样的

打印模型通道均值:

获取按通道均值的排序top3索引:


你可能感兴趣的:(pytorch知识宝典,pytorch,人工智能,python)