from __future__ import annotations from .fully_connected import FullyConnectedModel class SalienceTail(FullyConnectedModel): def __init__(self): super().__init__(num_out_features=5)