From fcb1f332e68771cb8eea356e3a2cc2cfdf429f9a Mon Sep 17 00:00:00 2001 From: huchukato Date: Fri, 28 Aug 2026 05:00:37 +0200 Subject: [PATCH] Re-export rife425 ONNX with 1080x1080 dummy input The previous 512x512 dummy input caused onnxsim to fix intermediate dimensions to 512px, which broke TensorRT engine build for the large resolution profile (720-1920px). Using 1080x1080 prevents this. Generated with [Devin](https://devin.ai) Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- export_onnx.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/export_onnx.py b/export_onnx.py index d010eab..8bdc6cc 100644 --- a/export_onnx.py +++ b/export_onnx.py @@ -139,9 +139,10 @@ def export_onnx(ckpt_name, ensemble, scale_factor): interpolation_model.load_state_dict(filtered_state_dict, strict=True) interpolation_model.eval().to(TORCH_DEVICE) - # # dummy data - img0 = torch.randn(1, 3, 512, 512).to(TORCH_DEVICE) - img1 = torch.randn(1, 3, 512, 512).to(TORCH_DEVICE) + # # dummy data — use 1080x1080 so onnxsim doesn't fix intermediate + # dimensions to 512px (which breaks TRT build for large profiles) + img0 = torch.randn(1, 3, 1080, 1080).to(TORCH_DEVICE) + img1 = torch.randn(1, 3, 1080, 1080).to(TORCH_DEVICE) timestep = torch.tensor([0.5], dtype=torch.float32).to(TORCH_DEVICE) # result = (interpolation_model(img0, img1, timestep)) @@ -179,7 +180,7 @@ def export_onnx(ckpt_name, ensemble, scale_factor): sim_model_path = os.path.join("models", f"{ckpt_name.split('.')[0]}_ensemble_{ensemble}_scale_{scale_factor}_sim.onnx") print("=> ONNX simplify start!") - sim_onnx_model, check = simplify(onnx_model) # convert(simplify) + sim_onnx_model, check = simplify(onnx_model) onnx.save(sim_onnx_model, sim_model_path) print("=> ONNX simplify done!") @@ -189,9 +190,9 @@ def export_onnx(ckpt_name, ensemble, scale_factor): print("=> sim ONNX Model check done!") -export_onnx(ckpt_name="rife47.pth", ensemble=True, scale_factor=1) -export_onnx(ckpt_name="rife49.pth", ensemble=True, scale_factor=1) -export_onnx(ckpt_name="rife417.pth", ensemble=True, scale_factor=1) export_onnx(ckpt_name="rife425.pth", ensemble=False, scale_factor=1) -export_onnx(ckpt_name="rife426.pth", ensemble=False, scale_factor=1) -export_onnx(ckpt_name="sudo_rife4_269.662_testV1_scale1.pth", ensemble=True, scale_factor=1) \ No newline at end of file +# export_onnx(ckpt_name="rife47.pth", ensemble=True, scale_factor=1) +# export_onnx(ckpt_name="rife49.pth", ensemble=True, scale_factor=1) +# export_onnx(ckpt_name="rife417.pth", ensemble=True, scale_factor=1) +# export_onnx(ckpt_name="rife426.pth", ensemble=False, scale_factor=1) +# export_onnx(ckpt_name="sudo_rife4_269.662_testV1_scale1.pth", ensemble=True, scale_factor=1) \ No newline at end of file