2025-09-24 02:39:51,831 [INFO ] [logger.py:dumpArgs:73]	 Starting with arguments
{"name": "celeba", "dataset": {"path": "../../datasets/CelebAMask/256/img", "lists_root": "datasets/celeba", "attributes_path": "../../datasets/CelebAMask/1024/CelebAMask-HQ-attribute-anno.txt"}, "batch_size": 250, "split": "train", "output": "./models/mean/", "verbose": 1, "cuda_index": 1}
2025-09-24 02:39:51,831 [INFO ] [learn_mean_imputator.py:<module>:35]	 Starting to learn mean for celeba
2025-09-24 02:39:51,831 [INFO ] [loader.py:getDatasetSplits:30]	 Loading CelebA dataset from ../../datasets/CelebAMask/256/img
2025-09-24 02:39:56,713 [INFO ] [celeba.py:createCelebADataset:139]	 Found 40 CelebA attributes for 30000 images: ['5_o_Clock_Shadow', 'Arched_Eyebrows', 'Attractive', 'Bags_Under_Eyes', 'Bald', 'Bangs', 'Big_Lips', 'Big_Nose', 'Black_Hair', 'Blond_Hair', 'Blurry', 'Brown_Hair', 'Bushy_Eyebrows', 'Chubby', 'Double_Chin', 'Eyeglasses', 'Goatee', 'Gray_Hair', 'Heavy_Makeup', 'High_Cheekbones', 'Male', 'Mouth_Slightly_Open', 'Mustache', 'Narrow_Eyes', 'No_Beard', 'Oval_Face', 'Pale_Skin', 'Pointy_Nose', 'Receding_Hairline', 'Rosy_Cheeks', 'Sideburns', 'Smiling', 'Straight_Hair', 'Wavy_Hair', 'Wearing_Earrings', 'Wearing_Hat', 'Wearing_Lipstick', 'Wearing_Necklace', 'Wearing_Necktie', 'Young']
2025-09-24 02:39:56,766 [INFO ] [celeba.py:createCelebADataset:149]	 Loading 26000 training images, 2000 validation images, and 2000 testing images
2025-09-24 02:39:56,804 [INFO ] [util.py:selectDevice:88]	 Using cuda:1 for tensor calculations
2025-09-24 02:39:57,155 [INFO ] [learn_mean_imputator.py:<module>:53]	 Using dataset split train
2025-09-24 02:39:57,155 [INFO ] [learn_mean_imputator.py:<module>:59]	 Started learning celeba mean
2025-09-24 02:39:57,540 [INFO ] [imputator.py:meanFromDataloader:126]	 Output shape: torch.Size([3, 256, 256])
2025-09-24 02:48:11,296 [INFO ] [imputator.py:meanFromDataloader:134]	 Computed dataset mean in 494.14113846980035 seconds
2025-09-24 02:48:11,297 [INFO ] [learn_mean_imputator.py:<module>:65]	 Saving mean to ./models/mean/celeba.pklz
