Conceptos básicos de poda y combate real (5): explicación detallada del código de poda

Para podar el modelo, solo podamos la capa con parámetros, usamos BatchNorm2d para canalizar la importancia $Parámetro\; \gamma$ para el entrenamiento de dilución. También se requiere poda para BatchNorm2d y sus capas frontal y trasera. Principalmente para capas con parámetros:Conv2d,BatchNorm2d,Linear. Pero no podaremos la capa Pool2d, porque Pool2d solo se usa para reducir la resolución y no hay parámetros que se puedan aprender ni procesamiento.

0 código de modelo VGG

import<