<torch.\_C.Generator at 0x7f662cd9eb58>train: (50000, 32, 32, 3)test: (10000, 32, 32, 3)('conv_layers', Sequential( (0): Conv2d(3, 8, kernel_size=