megengine.module.quantized.ConvBnRelu2d¶
-
class
ConvBnRelu2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, conv_mode='CROSS_CORRELATION', compute_mode='DEFAULT', dtype=None, **kwargs)[源代码]¶ 基类：
megengine.module.quantized.conv_bn._ConvBnActivation2d量化（quantized）版本
ConvBnRelu2d。方法
__init__(in_channels, out_channels, kernel_size)- 参数名称
模块名称，子类的
kwargs将会保存和传递此关键字参数。
apply(fn)对当前模块中的所有模块应用函数
fn，包括当前模块本身。buffers([recursive])返回该模块中对于buffers的一个可迭代对象。
calc_conv(inp, weight, bias)calc_conv_quantized(inp[, nonlinear_mode])children(**kwargs)返回一个可迭代对象，可遍历所有属于当前模块的直接属性的子模块。
disable_quantize([value])设置
module的quantize_diabled属性，并返回module。eval()当前模块中所有模块的
training属性（包括自身）置为False，并将其切换为推理模式。forward(inp)from_qat_module(qat_module)返回从
QATModule实例转换而来的QuantizedModule实例。load_state_dict(state_dict[, strict])加载一个参数字典，这个字典通常使用:func:`state_dict`得到。
modules(**kwargs)返回一个可迭代对象，可以遍历当前模块中的所有模块，包括其本身。
named_buffers([prefix, recursive])返回可遍历模块中 key 与 buffer 的键值对的可迭代对象，其中
key为从该模块至 buffer 的点路径（dotted path）。named_children(**kwargs)返回可迭代对象，可以遍历属于当前模块的直接属性的所有子模块（submodule）与键（key）组成的”key-submodule”对，其中’key’是子模块对应的属性名。
named_modules([prefix])返回可迭代对象，可以遍历当前模块包括自身在内的所有其内部模块所组成的key-module键-模块对，其中’key’是从当前模块到各子模块的点路径（dotted path）。
named_parameters([prefix, recursive])返回一个可迭代对象，可以遍历当前模块中key与
Parameter组成的键值对。其中key是从模块到Parameter的点路径（dotted path）。parameters([recursive])返回一个可迭代对象，遍历当前模块中的所有
Parameterregister_forward_hook(hook)给模块输出注册一个回调函数。
register_forward_pre_hook(hook)给模块输入注册一个回调函数。
replace_param(params, start_pos[, seen])用 param 替换模块的参数，被
ParamPack用来加速多机训练。reset_parameters()- 返回类型
state_dict([rst, prefix, keep_var])train([mode, recursive])当前模块中所有模块的
training属性（包括自身）置为mode。zero_grad()将所有参数的梯度置0。