code: def setup_function(_):
      global forward_calls
      forward_calls = 0
  
imports: import torch
  import torch.nn
  import pytest
  from allennlp.common.params import Params
  from allennlp.common.testing import multi_device
  from allennlp.nn.checkpoint.checkpoint_wrapper import CheckpointWrapper, TorchCheckpointWrapper
label_desc: Use image node_id 1 for calling a global function with example usage: setup_function(_) without return types