search

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

Albumentations Version #31

Open idinsmore1 opened 1 year ago

idinsmore1 commented 1 year ago

Hi there,

I'm trying to train on my own dataset, and when creating the environment from the environment.yml file, albumentations and it's specific version are not listed. When I install the package with conda post-environment creation, the training script breaks. What version of albumentations are you working with in this project? Thanks!

huberemanuel commented 1 year ago

I tried albumentations==1.2.1 and it worked well.