Method forward

tests/networks/utils/test_copy_model_state.py:31–34

(self, x)

        self.class_layer = torch.nn.Linear(n_m, n_class)

    def forward(self, x):
        x = self.layer(x)
        x = self.class_layer(x)
        return x


class _TestModelTwo(torch.nn.Module):