The shift function is not used

jeya-maria-jose / UNeXt-pytorch

Official Pytorch Code base for "UNeXt: MLP-based Rapid Medical Image Segmentation Network", MICCAI 2022

https://jeya-maria-jose.github.io/UNext-web/

MIT License

459 stars 76 forks source link

Closed raozhongyu closed 2 years ago

raozhongyu commented 2 years ago

Thanks for your great work, I notice that the function of shift in archs.py is not used.

jeya-maria-jose commented 2 years ago

Thank you for your observation. Yes, instead of calling the function; I just wrote it inside the forward function https://github.com/jeya-maria-jose/UNeXt-pytorch/blob/076937f1f0161f4baf8ffb5f2948ac67f93afe29/archs.py#L83. Sorry about that. Will comment out the function.