Method forward

tools/test_linear_probing_hf.py:181–182

(self, inputs)

        self.classifiers_dict.update(classifiers_dict)

    def forward(self, inputs) -> Dict[str, torch.Tensor]:
        return {k: v.forward(inputs) for k, v in self.classifiers_dict.items()}

    def __len__(self):
        return len(self.classifiers_dict)