2021年10月4日
pytorch基础知识--nn.Module()类
摘要： pytorch里面一切自定义操作基本上都是继承nn.Module类来实现的。 我们在定义自已的网络的时候，需要继承nn.Module类，并重新实现构造函数__init__构造函数和forward这两个方法。但有一些注意技巧： 一般把网络中具有可学习参数的层（如全连接层、卷积层等）放在构造函数__in 阅读全文
posted @ 2021-10-04 21:41 llllrj 阅读(136) 评论(0) 推荐(0) 编辑