code: def move_some_params_to_cuda(model, torch_model):
      model.embed.weight.data = model.embed.weight.cuda()
      torch_model.embed.weight.data = model.embed.weight.cuda()
      model.ln1.weight.data = model.ln1.weight.cuda()
      torch_model.ln1.weight.data = model.ln1.weight.cuda()
  
imports: import torch
  from colossalai.nn.optimizer import CPUAdam, HybridAdam
  from colossalai.testing import clear_cache_before_run, parameterize
  from tests.kit.model_zoo import model_zoo
label_desc: Use image node_id 1 for calling a global function with example usage: move_some_params_to_cuda(model, torch_model) without return types