2025-05-14 00:53:54,361 [INFO ] [logger.py:dumpArgs:73]	 Starting with arguments
{"name": "celeba", "dataset": {"path": "../../datasets/CelebAMask/256/img", "lists_root": "datasets/celeba", "targets": ["Blond_Hair"], "attributes_path": "../../datasets/CelebAMask/1024/CelebAMask-HQ-attribute-anno.txt"}, "masks": [{"name": "top"}, {"name": "bottom"}, {"name": "full"}, {"name": "none"}], "output": "./models/dirichlet-celeba/Blond_Hair/", "seed": 1337, "verbose": 1, "cuda_index": 0, "learning_rate": 0.0001, "training_iterations": 1000, "batch_size": 50, "validate_every": 10, "patience": 5, "evaluate_training": true, "teacher": "./models/celeba/Blond_Hair/celeba-20250508-133714.pklz", "masked_weight": 0.0, "dirichlet_weight": 1.0, "clean_weight": 0.0, "input": null, "architecture": {"name": "resnet_dirichlet", "activation": "exp"}}
2025-05-14 00:53:54,362 [INFO ] [learn_dirichlet_network.py:<module>:68]	 Starting to train celeba
2025-05-14 00:53:54,362 [INFO ] [loader.py:getDatasetSplits:23]	 Loading CelebA dataset from ../../datasets/CelebAMask/256/img
2025-05-14 00:53:59,378 [INFO ] [celeba.py:createCelebADataset:139]	 Found 1 CelebA attributes for 30000 images: ['Blond_Hair']
2025-05-14 00:53:59,433 [INFO ] [celeba.py:createCelebADataset:149]	 Loading 26000 training images, 2000 validation images, and 2000 testing images
2025-05-14 00:53:59,462 [INFO ] [util.py:selectDevice:85]	 Using cuda:0 for tensor calculations
2025-05-14 00:53:59,698 [INFO ] [learn_dirichlet_network.py:<module>:82]	 Constructing neural network
2025-05-14 00:53:59,698 [INFO ] [loader.py:_ensureNumClassesSet:22]	 Constructing ResNet Dirichlet with 1 targets
2025-05-14 00:54:00,090 [INFO ] [learn_dirichlet_network.py:<module>:105]	 Network has 11170434 parameters
2025-05-14 00:55:04,709 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 64.61707 seconds
2025-05-14 00:55:04,709 [INFO ] [learn_dirichlet_network.py:<module>:146]	 Initial training error 1.975655e+00
2025-05-14 00:55:09,623 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.91383 seconds
2025-05-14 00:55:09,623 [INFO ] [learn_dirichlet_network.py:<module>:150]	 Initial validation error 2.085318e+00
2025-05-14 00:55:09,623 [INFO ] [learn_dirichlet_network.py:<module>:153]	 Starting network learning
2025-05-14 00:56:30,293 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 1/1000 in 80.66421 seconds - error: -3.958273e+00
2025-05-14 00:56:30,293 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 00:56:30,293 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 1 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-1.pklz
2025-05-14 00:57:36,770 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 64.87280 seconds
2025-05-14 00:57:36,771 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -3.725821e+00
2025-05-14 00:57:41,787 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 5.01619 seconds
2025-05-14 00:57:41,787 [INFO ] [learn_dirichlet_network.py:<module>:213]	 Found new best model with error -3.822863e+00 < prev best 2.085318e+00
2025-05-14 00:59:02,098 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 2/1000 in 80.30497 seconds - error: -4.156132e+00
2025-05-14 01:00:21,487 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 3/1000 in 79.38959 seconds - error: -4.299523e+00
2025-05-14 01:01:36,515 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 4/1000 in 75.02791 seconds - error: -4.411710e+00
2025-05-14 01:02:51,372 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 5/1000 in 74.85668 seconds - error: -4.463268e+00
2025-05-14 01:04:06,297 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 6/1000 in 74.92467 seconds - error: -4.465805e+00
2025-05-14 01:05:21,400 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 7/1000 in 75.10290 seconds - error: -4.440450e+00
2025-05-14 01:06:36,496 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 8/1000 in 75.09587 seconds - error: -4.456412e+00
2025-05-14 01:07:51,420 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 9/1000 in 74.92350 seconds - error: -4.436525e+00
2025-05-14 01:09:06,483 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 10/1000 in 75.06322 seconds - error: -4.376886e+00
2025-05-14 01:10:21,200 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 11/1000 in 74.71712 seconds - error: -4.429938e+00
2025-05-14 01:10:21,201 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 01:10:21,201 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 11 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-11.pklz
2025-05-14 01:11:17,568 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 54.89740 seconds
2025-05-14 01:11:17,568 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -4.482943e+00
2025-05-14 01:11:21,803 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.23539 seconds
2025-05-14 01:11:21,803 [INFO ] [learn_dirichlet_network.py:<module>:213]	 Found new best model with error -4.494437e+00 < prev best -3.822863e+00
2025-05-14 01:12:37,046 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 12/1000 in 75.23796 seconds - error: -4.525794e+00
2025-05-14 01:13:52,278 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 13/1000 in 75.23121 seconds - error: -4.411403e+00
2025-05-14 01:15:07,469 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 14/1000 in 75.19084 seconds - error: -4.417007e+00
2025-05-14 01:16:22,496 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 15/1000 in 75.02694 seconds - error: -4.522445e+00
2025-05-14 01:17:37,727 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 16/1000 in 75.23055 seconds - error: -4.549335e+00
2025-05-14 01:18:52,723 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 17/1000 in 74.99643 seconds - error: -4.534991e+00
2025-05-14 01:20:08,072 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 18/1000 in 75.34849 seconds - error: -4.558734e+00
2025-05-14 01:21:23,594 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 19/1000 in 75.52199 seconds - error: -4.509570e+00
2025-05-14 01:22:39,033 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 20/1000 in 75.43877 seconds - error: -4.320118e+00
2025-05-14 01:23:54,433 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 21/1000 in 75.40025 seconds - error: -4.482635e+00
2025-05-14 01:23:54,433 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 01:23:54,434 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 21 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-21.pklz
2025-05-14 01:24:50,051 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 54.14560 seconds
2025-05-14 01:24:50,051 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -3.918082e+00
2025-05-14 01:24:54,197 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.14643 seconds
2025-05-14 01:24:54,197 [INFO ] [learn_dirichlet_network.py:<module>:206]	 Worsening on valid -3.935240e+00 > prev best -4.494437e+00
2025-05-14 01:26:09,289 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 22/1000 in 75.09117 seconds - error: -4.433382e+00
2025-05-14 01:27:24,535 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 23/1000 in 75.24647 seconds - error: -4.591932e+00
2025-05-14 01:28:39,884 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 24/1000 in 75.34878 seconds - error: -4.597218e+00
2025-05-14 01:29:54,996 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 25/1000 in 75.11152 seconds - error: -4.589209e+00
2025-05-14 01:31:10,240 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 26/1000 in 75.24395 seconds - error: -4.631313e+00
2025-05-14 01:32:25,457 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 27/1000 in 75.21725 seconds - error: -4.641418e+00
2025-05-14 01:33:40,708 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 28/1000 in 75.25024 seconds - error: -4.691783e+00
2025-05-14 01:34:55,748 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 29/1000 in 75.03958 seconds - error: -4.659350e+00
2025-05-14 01:36:10,718 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 30/1000 in 74.97078 seconds - error: -4.561811e+00
2025-05-14 01:37:25,663 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 31/1000 in 74.94479 seconds - error: -4.530873e+00
2025-05-14 01:37:25,663 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 01:37:25,664 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 31 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-31.pklz
2025-05-14 01:38:21,301 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 54.15972 seconds
2025-05-14 01:38:21,302 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -4.632946e+00
2025-05-14 01:38:25,455 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.15336 seconds
2025-05-14 01:38:25,455 [INFO ] [learn_dirichlet_network.py:<module>:213]	 Found new best model with error -4.545135e+00 < prev best -4.494437e+00
2025-05-14 01:39:39,908 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 32/1000 in 74.44725 seconds - error: -4.567184e+00
2025-05-14 01:40:54,628 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 33/1000 in 74.72061 seconds - error: -4.678892e+00
2025-05-14 01:42:09,448 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 34/1000 in 74.81930 seconds - error: -4.569813e+00
2025-05-14 01:43:24,271 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 35/1000 in 74.82257 seconds - error: -4.647023e+00
2025-05-14 01:44:39,080 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 36/1000 in 74.80930 seconds - error: -4.720017e+00
2025-05-14 01:45:53,527 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 37/1000 in 74.44713 seconds - error: -4.646758e+00
2025-05-14 01:47:08,176 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 38/1000 in 74.64821 seconds - error: -4.719654e+00
2025-05-14 01:48:22,880 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 39/1000 in 74.70405 seconds - error: -4.742197e+00
2025-05-14 01:49:37,658 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 40/1000 in 74.77761 seconds - error: -4.716023e+00
2025-05-14 01:50:52,431 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 41/1000 in 74.77288 seconds - error: -4.762504e+00
2025-05-14 01:50:52,431 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 01:50:52,431 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 41 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-41.pklz
2025-05-14 01:51:47,773 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 53.87475 seconds
2025-05-14 01:51:47,773 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -4.493978e+00
2025-05-14 01:51:51,927 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.15431 seconds
2025-05-14 01:51:51,927 [INFO ] [learn_dirichlet_network.py:<module>:206]	 Worsening on valid -4.324321e+00 > prev best -4.545135e+00
2025-05-14 01:53:06,623 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 42/1000 in 74.69516 seconds - error: -4.703430e+00
2025-05-14 01:54:21,400 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 43/1000 in 74.77696 seconds - error: -4.760249e+00
2025-05-14 01:55:36,161 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 44/1000 in 74.76137 seconds - error: -4.804653e+00
2025-05-14 01:56:50,492 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 45/1000 in 74.33022 seconds - error: -4.822944e+00
2025-05-14 01:58:05,132 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 46/1000 in 74.64041 seconds - error: -4.757848e+00
2025-05-14 01:59:19,691 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 47/1000 in 74.55820 seconds - error: -4.813904e+00
2025-05-14 02:00:34,399 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 48/1000 in 74.70812 seconds - error: -4.830695e+00
2025-05-14 02:01:49,212 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 49/1000 in 74.81286 seconds - error: -4.847073e+00
2025-05-14 02:03:03,821 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 50/1000 in 74.60939 seconds - error: -4.871992e+00
2025-05-14 02:04:18,172 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 51/1000 in 74.35051 seconds - error: -4.847246e+00
2025-05-14 02:04:18,172 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 02:04:18,172 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 51 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-51.pklz
2025-05-14 02:05:13,111 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 53.46720 seconds
2025-05-14 02:05:13,112 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -4.894547e+00
2025-05-14 02:05:17,203 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.09151 seconds
2025-05-14 02:05:17,203 [INFO ] [learn_dirichlet_network.py:<module>:206]	 Worsening on valid -4.128744e+00 > prev best -4.545135e+00
2025-05-14 02:06:31,524 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 52/1000 in 74.32083 seconds - error: -4.911529e+00
2025-05-14 02:07:45,877 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 53/1000 in 74.35237 seconds - error: -4.944503e+00
2025-05-14 02:09:00,327 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 54/1000 in 74.45012 seconds - error: -4.955595e+00
2025-05-14 02:10:15,002 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 55/1000 in 74.67455 seconds - error: -4.979661e+00
2025-05-14 02:11:29,619 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 56/1000 in 74.61703 seconds - error: -4.935215e+00
2025-05-14 02:12:44,081 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 57/1000 in 74.46201 seconds - error: -4.956860e+00
2025-05-14 02:13:58,447 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 58/1000 in 74.36571 seconds - error: -5.030050e+00
2025-05-14 02:15:12,714 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 59/1000 in 74.26739 seconds - error: -5.051620e+00
2025-05-14 02:16:27,330 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 60/1000 in 74.61530 seconds - error: -5.081760e+00
2025-05-14 02:17:41,961 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 61/1000 in 74.63137 seconds - error: -5.067412e+00
2025-05-14 02:17:41,962 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 02:17:41,962 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 61 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-61.pklz
2025-05-14 02:18:37,200 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 53.77208 seconds
2025-05-14 02:18:37,200 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -5.154050e+00
2025-05-14 02:18:41,443 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.24283 seconds
2025-05-14 02:18:41,443 [INFO ] [learn_dirichlet_network.py:<module>:206]	 Worsening on valid -3.911980e+00 > prev best -4.545135e+00
2025-05-14 02:19:55,827 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 62/1000 in 74.38366 seconds - error: -5.140824e+00
2025-05-14 02:21:10,323 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 63/1000 in 74.49593 seconds - error: -5.162722e+00
2025-05-14 02:22:24,882 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 64/1000 in 74.55947 seconds - error: -5.164766e+00
2025-05-14 02:23:39,383 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 65/1000 in 74.50033 seconds - error: -5.213564e+00
2025-05-14 02:24:54,286 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 66/1000 in 74.90285 seconds - error: -5.210465e+00
2025-05-14 02:26:09,030 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 67/1000 in 74.74427 seconds - error: -5.262387e+00
2025-05-14 02:27:23,568 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 68/1000 in 74.53707 seconds - error: -5.320114e+00
2025-05-14 02:28:38,014 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 69/1000 in 74.44587 seconds - error: -5.315063e+00
2025-05-14 02:29:52,146 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 70/1000 in 74.13179 seconds - error: -5.272580e+00
2025-05-14 02:31:06,591 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 71/1000 in 74.44486 seconds - error: -5.350966e+00
2025-05-14 02:31:06,591 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 02:31:06,591 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 71 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-71.pklz
2025-05-14 02:32:01,577 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 53.52562 seconds
2025-05-14 02:32:01,577 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -5.340912e+00
2025-05-14 02:32:05,686 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.10941 seconds
2025-05-14 02:32:05,687 [INFO ] [learn_dirichlet_network.py:<module>:206]	 Worsening on valid 2.895373e+00 > prev best -4.545135e+00
2025-05-14 02:33:20,406 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 72/1000 in 74.71932 seconds - error: -5.352516e+00
2025-05-14 02:34:34,756 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 73/1000 in 74.35024 seconds - error: -5.376800e+00
2025-05-14 02:35:49,032 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 74/1000 in 74.27527 seconds - error: -5.415919e+00
2025-05-14 02:37:03,685 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 75/1000 in 74.65279 seconds - error: -5.434531e+00
2025-05-14 02:38:18,258 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 76/1000 in 74.57260 seconds - error: -5.470737e+00
2025-05-14 02:39:32,843 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 77/1000 in 74.58506 seconds - error: -5.467167e+00
2025-05-14 02:40:47,095 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 78/1000 in 74.25177 seconds - error: -5.496836e+00
2025-05-14 02:42:01,281 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 79/1000 in 74.18671 seconds - error: -5.507180e+00
2025-05-14 02:43:15,873 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 80/1000 in 74.59114 seconds - error: -5.518355e+00
2025-05-14 02:44:30,517 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 81/1000 in 74.64425 seconds - error: -5.529480e+00
2025-05-14 02:44:30,517 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 02:44:30,518 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 81 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-81.pklz
2025-05-14 02:45:26,492 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 54.50706 seconds
2025-05-14 02:45:26,492 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -5.531705e+00
2025-05-14 02:45:30,634 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.14127 seconds
2025-05-14 02:45:30,634 [INFO ] [learn_dirichlet_network.py:<module>:206]	 Worsening on valid 1.272878e+01 > prev best -4.545135e+00
2025-05-14 02:46:44,643 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 82/1000 in 74.00877 seconds - error: -5.565845e+00
2025-05-14 02:47:59,364 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 83/1000 in 74.72073 seconds - error: -5.583955e+00
2025-05-14 02:49:15,804 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 84/1000 in 76.43983 seconds - error: -5.585930e+00
2025-05-14 02:50:31,773 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 85/1000 in 75.96896 seconds - error: -5.461159e+00
2025-05-14 02:51:47,653 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 86/1000 in 75.87962 seconds - error: -5.608061e+00
2025-05-14 02:53:03,175 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 87/1000 in 75.52204 seconds - error: -5.594581e+00
2025-05-14 02:54:18,583 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 88/1000 in 75.40832 seconds - error: -5.642052e+00
2025-05-14 02:55:34,446 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 89/1000 in 75.86255 seconds - error: -5.682753e+00
2025-05-14 02:56:50,567 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 90/1000 in 76.12075 seconds - error: -5.652578e+00
2025-05-14 02:58:06,485 [INFO ] [learn_dirichlet_network.py:<module>:185]	 celeba iteration 91/1000 in 75.91848 seconds - error: -5.665554e+00
2025-05-14 02:58:06,486 [INFO ] [learn_dirichlet_network.py:<module>:190]	 Evaluating the model via validation data
2025-05-14 02:58:06,486 [INFO ] [learn_dirichlet_network.py:<module>:195]	 Saving model at iteration 91 to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354-91.pklz
2025-05-14 02:59:02,869 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 54.92385 seconds
2025-05-14 02:59:02,870 [INFO ] [learn_dirichlet_network.py:<module>:200]	 Training error in evaluate mode -5.741144e+00
2025-05-14 02:59:07,080 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.21003 seconds
2025-05-14 02:59:07,080 [INFO ] [learn_dirichlet_network.py:<module>:206]	 Worsening on valid 1.180198e+01 > prev best -4.545135e+00
2025-05-14 02:59:07,080 [INFO ] [learn_dirichlet_network.py:<module>:208]	 Exceeding patience 5, stopping training
2025-05-14 02:59:07,080 [INFO ] [learn_dirichlet_network.py:<module>:220]	 Network learning done in 7437.45675 secs
2025-05-14 02:59:07,095 [INFO ] [learn_dirichlet_network.py:<module>:229]	 Saving model to ./models/dirichlet-celeba/Blond_Hair/celeba-20250514-005354.pklz
2025-05-14 03:00:04,335 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 55.77870 seconds
2025-05-14 03:00:04,336 [INFO ] [learn_dirichlet_network.py:<module>:239]	 Loss for train is -4.627656936645508:
tensor([-4.6277])
2025-05-14 03:00:08,608 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.27188 seconds
2025-05-14 03:00:08,608 [INFO ] [learn_dirichlet_network.py:<module>:239]	 Loss for validate is -4.545135021209717:
tensor([-4.5451])
2025-05-14 03:00:12,894 [INFO ] [regressor.py:evaluateData:67]	 Evaluated regressor with data loader in 4.28560 seconds
2025-05-14 03:00:12,894 [INFO ] [learn_dirichlet_network.py:<module>:239]	 Loss for test is -4.505692958831787:
tensor([-4.5057])
