benchmarks/torch_ctl_benchmark/dense_model_benchmark.py/TorchModel/__init__
class TorchModel:    def __init__(self):
        super().__init__()

        self.dense1 = torch.nn.Linear(8192, 64)
        self.activation1 = torch.nn.ReLU()
        self.dense2 = torch.nn.Linear(64, 8)
        self.activation2 = torch.nn.ReLU()
        self.dense3 = torch.nn.Linear(8, num_classes)
        self.softmax = torch.nn.Softmax(dim=1)