code: def test_attention_model():
      params = [4, 4, 8, 8, 16]
      model = ResNet152(params, att=True)
      _ = model(sample_input)
  
file_path: KD_Lib/tests/test_models.py
imports: import torch
  from KD_Lib.models import LeNet, LSTMNet, ModLeNet, NetworkInNetwork, ResNet18, ResNet34, ResNet50, ResNet101, ResNet152, Shallow
label_desc: Use image node_id 2 for calling a global function with example usage: test_attention_model() without return types