2025-09-24 02:41:44,131 [INFO ] [logger.py:dumpArgs:73]	 Starting with arguments
{"name": "mnist", "dataset": {"path": "../../datasets/mnist", "image_size": 224}, "batch_size": 50, "split": "train", "output": "./models/mean/", "verbose": 1, "cuda_index": 1}
2025-09-24 02:41:44,131 [INFO ] [learn_mean_imputator.py:<module>:35]	 Starting to learn mean for mnist
2025-09-24 02:41:44,131 [INFO ] [loader.py:getDatasetSplits:24]	 Loading MNIST dataset from ../../datasets/mnist
2025-09-24 02:41:44,185 [INFO ] [baselines.py:createMNISTDataset:69]	 Using 10 MNIST targets: ['0 - zero', '1 - one', '2 - two', '3 - three', '4 - four', '5 - five', '6 - six', '7 - seven', '8 - eight', '9 - nine']
2025-09-24 02:41:44,200 [INFO ] [baselines.py:getTransform:24]	 Resizing images from 28 to 224
2025-09-24 02:41:46,689 [INFO ] [torch_utils.py:split:72]	 Split training from 60000 into 42000 training images and 18000 validation images. Found 10000 testing images.
2025-09-24 02:41:46,724 [INFO ] [util.py:selectDevice:88]	 Using cuda:1 for tensor calculations
2025-09-24 02:41:46,878 [INFO ] [learn_mean_imputator.py:<module>:53]	 Using dataset split train
2025-09-24 02:41:46,878 [INFO ] [learn_mean_imputator.py:<module>:59]	 Started learning mnist mean
2025-09-24 02:41:46,953 [INFO ] [imputator.py:meanFromDataloader:126]	 Output shape: torch.Size([1, 224, 224])
2025-09-24 02:42:01,963 [INFO ] [imputator.py:meanFromDataloader:134]	 Computed dataset mean in 15.084947464987636 seconds
2025-09-24 02:42:01,964 [INFO ] [learn_mean_imputator.py:<module>:65]	 Saving mean to ./models/mean/mnist.pklz
