摘要: 模型构造 基于Block类的模型构造方法:它让模型构造更加灵活。 继承Block类来构造模型 Block类是nn模块里提供的一个模型构造类,我们可以继承它来定义我们想要的模型。下面继承Block类构造本节开头提到的多层感知机。这里定义的MLP类重载了Block类的__init__函数和forward 阅读全文
posted @ 2021-02-17 14:53 strategist_614 阅读(89) 评论(0) 推荐(0) 编辑