2020年8月25日

torch.nn.Sequential()搭建神经网络模型

摘要: 一、第一种方式(可以配合一些条件判断语句动态添加) 模板——torch.nn.Sequential()的一个对象.add_module(name, module)。 name:某层次的名字;module:需要添加的子模块,如卷积、激活函数等等。 添加子模块到当前模块中。 可以通过 name 属性来访 阅读全文

posted @ 2020-08-25 10:40 cltt 阅读(1850) 评论(0) 推荐(0) 编辑

导航