bonlime / pytorch-tools

Tool box for PyTorch
MIT License
186 stars 16 forks source link

add bit-M-ResNet #67

Closed bonlime closed 4 years ago

bonlime commented 4 years ago

1) Скопировал код для моделей от гугла 2) добавил автоматичекую подгрузку весов (это удобно) 3) добавил создание моделей через функции

они созданы исключительно для transfer learning, поэтому там многие параметры не поддерживаются пока. потом добавлю возможность использовать как энкодер