code: def check_state_dict_fp32(self, state_dict):
      for key in state_dict:
          if "num_batches_tracked" in key:
              continue
          param = state_dict[key]
          self.assertEqual(
              param.type(), FLOAT, "Parameter in state_dict not FLOAT"
          )
  
imports: import unittest
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from apex import amp
  from utils import common_init, FLOAT
label_desc: Use image node_id 3 for calling the TestCheckpointing obj's underlying member method code with example usage: obj.check_state_dict_fp32(state_dict) without return types