Исправления для нейросетей
Created by: KofTD
Densenet и MobilenetV2 имеют странную причуду: после свёрток использовать функциональный вызов adaptive_avg_pooling'а в forward'e. Это и вызывало проблемы при срезах, цепляющих их классификатор. Я не нашёл каких-либо объяснений такому решению, учитывая полную идентичность класса AdaptiveAvgPool2d и вызова функции с схожим именем.
В остальном исправления должны сделать все сети готовыми к работе