Compatibility Inquiry - Support for PyTorch 2.4.1?

[songyuc]

Hello,

I am a PhD student studying the Swin UNETR model and attempting to reproduce your work in this open-source project. I noticed that the latest stable version of PyTorch is 2.4.1, and I would like to confirm if your project supports this version or if there are specific requirements for PyTorch versions.

If there are any recommended settings or considerations, please let me know. Additionally, if I can assist in any way to facilitate this compatibility, please feel free to reach out.

Thank you for your time and effort,
Yucheng Song