以下是我将 pth 导出为 onnx 的步骤（仅作为自己未来的参考）：

1. 打开 Colab notebook 并点击 Runtime > Run All。
2. 在 Colab 编辑器中打开 main_test_swinir.py，并在 output = model(img_lq) 这一行之后添加以下代码：

torch.onnx.export(model, img_lq, \"003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN-dynamic.onnx\", export_params=True, opset_version=12, do_constant_folding=True, verbose=True, input_names = ['input'], output_names = ['output'], dynamic_axes={'input' : {2 : 'h', 3 : 'w'}, 'output' : {2 : 'h', 3 : 'w'}})

3. 运行以下命令：

!python SwinIR/main_test_swinir.py --task real_sr --model_path experiments/pretrained_models/003_realSR_BSRGAN_DFO_s64w8_SwinIR-M_x4_GAN.pth --folder_lq BSRGAN/testsets/RealSRSet --scale 4

然后你就会在本仓库中看到生成的 ONNX 文件。