zoukankan      html  css  js  c++  java
  • 打印pytorch每一层的输入输出

    打印pytorch每层参数。

    采用的是直接在层中加入txt的写入。需要修改的文件位置:./site-packages/torch/nn/modules/

    Conv2D

            v = F.conv2d(input, self.weight, self.bias, self.stride,self.padding, self.dilation, self.groups)

            v = v.detach()

            v = v.cpu().numpy().flatten()

            fp = open('conv2d.txt', 'w')

            for d in v:

                fp.write(str(d)+' ')

            print('conv2d has Convert Completely!') 

            fp.close()  

    输入数据input

            v = label.detach()

            v = v.cpu().numpy().flatten()

            fp = open('input.txt', 'w')

            for d in v:

                fp.write(str(d)+' ')

            print('input has Convert Completely!') 

            fp.close()  

    其它层仿照上面的就可以了。

  • 相关阅读:
    [转载]qemu-kvm安装配置
    Hadoop通过c语言API访问hdfs
    hadoop和hdfs环境搭建
    OpenCV installation for Ubuntu 12.04
    homework-01
    linux命令2
    压缩tar
    anaconda 安装opencv
    anconda安装第三方库
    开源代码
  • 原文地址:https://www.cnblogs.com/wjjcjj/p/13890058.html
Copyright © 2011-2022 走看看