bonlime / pytorch-tools

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

redo interface #16

Closed bonlime closed 5 years ago

bonlime commented 5 years ago

@zakajd Посмотри. Передел то, как оборачиваются модели. Теперь сохраняется сигнатура вызова. т.е. вызывая каждую из мини-функций ты видишь все аргументы главного класса и соответствующий docstring. Стало максимально удобно.

bonlime commented 5 years ago

Я пока только для vgg добавил, но для резнетов по аналогии делается

bonlime commented 5 years ago

Вообще этот код получается даже явнее чем код в torchvision.models, потому что там не прокидываются доки и о синтаксисе можно только догадываться

bonlime commented 5 years ago

а сейчас я осознал что засунуть pretrain внутрь на самом деле не очень хорошая идея, потому что вызов условной функции vgg11_bn задает основную архитектуру, но кроме этого там есть еще параметры, которые могут делать веса обученные на imagenet неприменимыми. (InplaceABN например требует переобучения). В идеале хотелось бы пытаться по аргументам вызова подгрузить правильные веса и ругаться если их нет, но это раздует каждую из функций