Перейти к содержанию

Index

5.5 LinearClassifierHead (опционально)

Для многоклассовых и пользовательских сценариев (cnn_neat/heads.py):

\[ \mathbf{z}_b = \mathbf{W}_\text{fc} \cdot \text{flatten}(\mathbf{F}_b) + \mathbf{b}_\text{fc} \in \mathbb{R}^{K} \]

где \(\text{flatten}\) имеет размерность \(3 \cdot H_\text{out} \cdot W_\text{out}\), \(K\) — число классов. При несовпадении spatial size — предварительный AdaptiveAvgPool2d.

Не используется в стандартном CIFAR-10 binary pipeline.