hunk.code_window: @pytest.mark.parametrize('batch_size', [2])
   def test_model_backward(model_name, batch_size):
       """Run a single forward pass with each model"""
       model = create_model(model_name, pretrained=False, num_classes=42)
       num_params = sum([x.numel() for x in model.parameters()])
       model.train()
  
hunk.labels: keep keep add keep keep keep keep
hunk.after_edit:     input_size = _get_input_size(model_name=model_name, target=TARGET_BWD_SIZE)
       if max(input_size) > MAX_BWD_SIZE:
           pytest.skip("Fixed input size model > limit.")
  
hunk.file_path: tests/test_models.py
hunk.type: add
file: model-index==0.1.10
  jinja2==2.11.3
  