Closed bonlime closed 5 years ago
Я пока только для vgg добавил, но для резнетов по аналогии делается
Вообще этот код получается даже явнее чем код в torchvision.models
, потому что там не прокидываются доки и о синтаксисе можно только догадываться
а сейчас я осознал что засунуть pretrain внутрь на самом деле не очень хорошая идея, потому что вызов условной функции vgg11_bn
задает основную архитектуру, но кроме этого там есть еще параметры, которые могут делать веса обученные на imagenet неприменимыми. (InplaceABN например требует переобучения). В идеале хотелось бы пытаться по аргументам вызова подгрузить правильные веса и ругаться если их нет, но это раздует каждую из функций
@zakajd Посмотри. Передел то, как оборачиваются модели. Теперь сохраняется сигнатура вызова. т.е. вызывая каждую из мини-функций ты видишь все аргументы главного класса и соответствующий docstring. Стало максимально удобно.