Fixed Reverse Distillation export to ONNX

[adrianboguszewski]

📝 Description

• Replace kornia gaussian filter which fails when exporting to ONNX with a custom one
• 🛠️ Fixes 🐞 ONNX Export for Reverse Distillation and RKDE fails #1513

✨ Changes

Select what type of change your PR is:

• 🐞 Bug fix (non-breaking change which fixes an issue)
• 🔨 Refactor (non-breaking change which refactors the code base)
• 🚀 New feature (non-breaking change which adds functionality)
• 💥 Breaking change (fix or feature that would cause existing functionality to not work as expected)
• 📚 Documentation update
• 🔒 Security update

✅ Checklist

Before you submit your pull request, please make sure you have completed the following steps:

• 📋 I have summarized my changes in the CHANGELOG and followed the guidelines for my type of change (skip for minor changes, documentation updates, and test enhancements).
• 📚 I have made the necessary updates to the documentation (if applicable).
• 🧪 I have written tests that support my changes and prove that my fix is effective or my feature works (if applicable).

For more information about code review checklists, see the Code Review Checklist.

[samet-akcay]

Thanks for addressing this!

[samet-akcay]

The tests are failing due to device mismatch somewhere

[adrianboguszewski]

Let's see if it helps. I cannot test it locally due to a lack of a good GPU