megengine.module.AvgPool2d.train

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

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

参数

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

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

返回类型

None

上一页

megengine.module.AvgPool2d.tensors

下一页

megengine.module.AvgPool2d.zero_grad