search

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

AttributeError: 'ByobNet' object has no attribute 'global_pool' #30

Open MichaelMonashev opened 2 years ago

MichaelMonashev commented 2 years ago

В timm есть очень быстрые сетки семейства gernet_[s,m,l] взятые отсюда https://github.com/idstcv/GPU-Efficient-Networks . Они идеально подходят для поиска аугов при классификации, ибо быстрее. Например, gernet_s быстрее resnet18 примерно вдвое и даёт большую точность. Но использовать сейчас gernet нельзя из-за ошибки, возникающей при старте поиска. AttributeError: 'ByobNet' object has no attribute 'global_pool'

Вот код gernet-ов: https://github.com/rwightman/pytorch-image-models/blob/master/timm/models/byobnet.py

yustiks commented 2 years ago

В timm есть очень быстрые сетки семейства gernet_[s,m,l] взятые отсюда https://github.com/idstcv/GPU-Efficient-Networks . Они идеально подходят для поиска аугов при классификации, ибо быстрее. Например, gernet_s быстрее resnet18 примерно вдвое и даёт большую точность. Но использовать сейчас gernet нельзя из-за ошибки, возникающей при старте поиска. AttributeError: 'ByobNet' object has no attribute 'global_pool'

Вот код gernet-ов: https://github.com/rwightman/pytorch-image-models/blob/master/timm/models/byobnet.py

аналогичная проблема. Получилось решить?