Remove warnings from utils.py.

aidge_interop_torch/utils.py

@@ -217,7 +217,9 @@ class ContextNoBatchNormFuse:
                     else:
                         eaf = current_bn.momentum
                     assert current_bn.num_batches_tracked is not None
-                    current_bn.num_batches_tracked.add_(1)
+                    # Old way, raise a warning
+                    # current_bn.num_batches_tracked.add_(1)
+                    current_bn.num_batches_tracked = current_bn.num_batches_tracked + 1
                     if current_bn.momentum is None:  # use cumulative moving average
                         eaf = 1.0 / current_bn.num_batches_tracked.item()
                     else:  # use exponential moving average
@@ -240,11 +242,14 @@ class ContextNoBatchNormFuse:
                         eaf,                     # exponential_average_factor
                         current_bn.eps,          # epsilon
                     )
+                    # Old way, raise a warning
+                    # current_bn.running_mean.copy_(torch.nn.Parameter(
+                    #     saved_run_mean).requires_grad_(False))
+                    # current_bn.running_var.copy_(torch.nn.Parameter(
+                    #     saved_run_var).requires_grad_(False))
+                    current_bn.running_mean = saved_run_mean
+                    current_bn.running_var = saved_run_var
 
-                    current_bn.running_mean.copy_(torch.nn.Parameter(
-                        saved_run_mean).requires_grad_(False))
-                    current_bn.running_var.copy_(torch.nn.Parameter(
-                        saved_run_var).requires_grad_(False))
                     if current_bn.affine:
                         current_bn.bias = (torch.nn.Parameter(saved_bias))
                         current_bn.weight = (torch.nn.Parameter(saved_weight))