code: def main():
      old_w = torch.load("u2net.pth", map_location="cpu")
      new_m = u2net_full()
  
      # old_w = torch.load("u2netp.pth", map_location='cpu')
      # new_m = u2net_lite()
  
      w = convert(old_w)
      new_m.load_state_dict(w, strict=True)
      torch.save(new_m.state_dict(), "u2net_full.pth")
  
imports: import re
  import torch
  from src import u2net_full, u2net_lite
label_desc: Use image node_id 4 for calling a global function with example usage: main() without return types