Method forward

tests/test_linear4bit.py:500–503

(self, x)

            )

        def forward(self, x):
            for layer in self.layers:
                x = torch.utils.checkpoint.checkpoint(layer, x, use_reentrant=False)
            return x

    net = CheckpointedNet().to(device)