code: def main_1():
      from u2net import U2NET, U2NETP
  
      old_m = U2NET()
      old_m.load_state_dict(torch.load("u2net.pth", map_location="cpu"))
      new_m = u2net_full()
  
      # old_m = U2NETP()
      # old_m.load_state_dict(torch.load("u2netp.pth", map_location='cpu'))
      # new_m = u2net_lite()
  
      old_w = old_m.state_dict()
  
      w = convert(old_w)
      new_m.load_state_dict(w, strict=True)
  
      torch.random.manual_seed(0)
      x = torch.randn(1, 3, 288, 288)
      old_m.eval()
      new_m.eval()
      with torch.no_grad():
          out1 = old_m(x)[0]
          out2 = new_m(x)
          assert torch.equal(out1, out2)
          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 3 for calling a global function with example usage: main_1() without return types