batch norm explicit dim fix
@@ -19,7 +19,7 @@ def convert_batch_norm_trt7(ctx):
bias = bias.detach().cpu().numpy() - running_mean.detach().cpu().numpy() * scale
power = np.ones_like(scale)
- layer = ctx.network.add_scale_nd(input_trt, trt.ScaleMode.CHANNEL, bias, scale, power, 0)
+ layer = ctx.network.add_scale_nd(input_trt, trt.ScaleMode.CHANNEL, bias, scale, power, 1)
output._trt = layer.get_output(0)
