code: def convert_head(
      model_key, model_weight, state_dict, converted_names
  ):
      new_key = model_key.replace("head.fc", "fc")
      state_dict[new_key] = model_weight
      converted_names.add(model_key)
      print(f"Convert {model_key} to {new_key}")
  
imports: import argparse
  from collections import OrderedDict
  import torch
label_desc: Use image node_id 2 for calling a global function with example usage: convert_head(model_key, model_weight, state_dict, converted_names) without return types