text: Title: mask2former模型使用pytorch2torchscript.py，导出为gpu上的torchscript模型后，精度损失很厉害，而cpu的不会
  Body: 我也尝试了直接使用下面代码来导出模型，和上面一样都没有报错，但结果依然没有变化
  
  `model = init_model(config_path, checkpoint_path, device='cuda:0')
  
  verify = True
  imgs = torch.randn(1,3,512,512).to("cuda")
  traced_model = torch.jit.trace(model,example_inputs=imgs,check_trace=verify,)
  traced_model.save(output_file)`
  
  而使用result = mmseg.apis.inference_model(model, img)来推理，精度是正常的，**使用cpu的torchscript模型监督也正常**
  
  **是不是mmsegmentation对于导出gpu上的torchscript模型有bug，不适配**
  
  我想使用traced_model = torch.jit.script(model)来导出，有些报错一下解决不了
  