code: def bn2float(module:nn.Module)->nn.Module:
      "If `module` is batchnorm don't use half precision."
      if isinstance(module, torch.nn.modules.batchnorm._BatchNorm): module.float()
      for child in module.children(): bn2float(child)
      return modulf bn2float(module:nn.Module)->nn.Module:
      "If `module` is batchnorm don't use half precision."
      if isinstance(module, torch.nn.modules.batchnorm._BatchNorm): module.float()
      for child in module.children(): bn2float(child)
      return module
docs: If `module` is batchnorm don't use half precision.