code: def test_save_load_batch_norm(self):
      state_dim = 8
      action_dim = 4
      model = FullyConnectedCritic(
          state_dim,
          action_dim,
          sizes=[8, 4],
          activations=["relu", "relu"],
          use_batch_norm=True,
      )
      # Freezing batch_norm
      model.eval()
      self.check_save_load(model)
  
file_path: ReAgent/reagent/test/models/test_critic.py
imports: import logging
  import unittest
  import torch
  import torch.nn
  from reagent.core import types
  from reagent.models.critic import FullyConnectedCritic
  from reagent.test.models.test_utils import run_model_jit_trace
label_desc: Use image node_id 4 for calling the TestFullyConnectedCritic obj's underlying member method code with example usage: obj.test_save_load_batch_norm() without return types