torch.manual_seed(7)
model = TinyClassifier().eval()
sample = torch.randn(4, 8)