code: def test_qat_quantization():
      model = models.quantization.resnet18(quantize=False)
      model.fc.out_features = 10
  
      optimizer = torch.optim.Adam(model.parameters())
      quantizer = QAT_Quantizer(
          model, train_loader, test_loader, optimizer
      )
      _ = quantizer.quantize(1, 1, 1, 1)
      quantizer.get_model_sizes()
      quantizer.get_performance_statistics()
  
file_path: KD_Lib/tests/test_quantization.py
imports: from copy import deepcopy
  import torch
  from torch.utils.data import DataLoader
  from torchvision import models
  from KD_Lib.Quantization import Dynamic_Quantizer, QAT_Quantizer, Static_Quantizer
  from .utils import MockImageClassifier, MockVisionDataset
label_desc: Use image node_id 3 for calling a global function with example usage: test_qat_quantization() without return types