megengine.module.qat.ConvBn2d.update_running_mean_and_running_var

ConvBn2d.update_running_mean_and_running_var(bn_mean, bn_var, num_elements_per_channel)

上一页

megengine.module.qat.ConvBn2d.train

下一页

megengine.module.qat.ConvBn2d.zero_grad