megengine.module.Conv1d.train

Conv1d.train(mode=True, recursive=True)

将该模块中的所有模块(包括它自身)的训练模式设置为 mode 。 可便捷地将这些模块的 training 属性设置为 mode ，但仅对某些模块有效(例如 BatchNorm2d, Dropout, Observer)

参数

• mode (bool) – 为模块设置的训练模式。

• recursive (bool) – 是否要递归调用子模块的 train() 。

返回类型

None

上一页

megengine.module.Conv1d.tensors

下一页

megengine.module.Conv1d.zero_grad