albumentations-team / autoalbument

AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - https://albumentations.ai/docs/autoalbument/
https://albumentations.ai/docs/autoalbument/
MIT License
198 stars 20 forks source link

Update requirements #29

Open MichaelMonashev opened 2 years ago

MichaelMonashev commented 2 years ago

Сейчас с библиотекой ставятся очень старые версии timm и segmentation-models-pytorch. Не уверен, что нужно фиксировать версию timm, ибо либа довольно стабильная и если что, segmentation-models-pytorch поставит нужную версию. Сейчас версия timm такая же, как в https://github.com/qubvel/segmentation_models.pytorch/blob/master/requirements.txt .