Hi!
First of all, thank you very for your work!
Do you plan to release the pre-trained model weights? They would be incredibly helpful for further exploration.
Also, could you provide more specific details on reproducing your results? Specifically, details on training time (hardware, duration) and memory requirements.