megengine.module.qat.ConvBnRelu2d.update_running_mean_and_running_var

ConvBnRelu2d.update_running_mean_and_running_var(bn_mean, bn_var, num_elements_per_channel)

上一页

megengine.module.qat.ConvBnRelu2d.train

下一页

megengine.module.qat.ConvBnRelu2d.zero_grad