Arch: resnet50_pt
Bs trn: 128
Bs val: 128
Hidden dim: 256
Dataset: celebA
Resample class: 
Slice with: rep
Rep cluster method: gmm
Num anchor: 32
Num positive: 32
Num negative: 32
Num negative easy: 0
Weight anc by loss: False
Weight pos by loss: False
Weight neg by loss: False
Anc loss temp: 0.5
Pos loss temp: 0.5
Neg loss temp: 0.5
Data wide pos: False
Target sample ratio: 1
Balance targets: False
Additional negatives: False
Hard negative factor: 0
Full contrastive: False
Train encoder: False
No projection head: False
Projection dim: 128
Batch factor: None
Temperature: 0.05
Single pos: False
Supervised linear scale up: False
Supervised update delay: 0
Contrastive weight: 0.5
Classifier update interval: 8
Optim: sgd
Max epoch: 5
Lr: 0.0001
Momentum: 0.9
Weight decay: 0.1
Weight decay c: 0.1
Stopping window: 30
Load encoder: 
Freeze encoder: False
Finetune epochs: 0
Clip grad norm: False
Lr scheduler classifier: 
Lr scheduler: 
Grad clip grad norm: False
Erm: False
Erm only: False
Pretrained spurious path: 
Max epoch s: 1
Bs trn s: 32
Lr s: 0.001
Momentum s: 0.9
Weight decay s: 0.0005
Slice temp: 10
Log loss interval: 10
Checkpoint interval: 50
Grad checkpoint interval: 50
Log visual interval: 100
Log grad visual interval: 50
Verbose: True
Seed: 36
Replicate: 0
No cuda: False
Resume: False
New slice: False
Num workers: 12
Evaluate: False
Data cmap: hsv
Test cmap: 
P correlation: 0.9
P corr by class: None
Train classes: ['blond', 'nonblond']
Train class ratios: None
Test shift: random
Flipped: False
Q: 0.7
Pretrained bmodel: False
Cosine: False
Exp: stage_one_erm
Supervised contrast: True
Prioritize spurious pos: False
Contrastive type: cnc
Compute auroc: False
Model type: resnet50_pt_cnc
Criterion: cross_entropy
Pretrained: False
Max grad norm: 1.0
Adam epsilon: 1e-08
Warmup steps: 0
Max grad norm s: 1.0
Adam epsilon s: 1e-08
Warmup steps s: 0
Grad max grad norm: 1.0
Grad adam epsilon: 1e-08
Grad warmup steps: 0
Device: cuda
Img file type: .png
Display image: False
Image path: ./images/celebA/celebA/config/contrastive_umaps
Log interval: 1
Log path: ./logs/celebA/config
Results path: ./results/celebA/config
Model path: ./model/celebA/config
Loss factor: 1
Supersample labels: False
Subsample labels: False
Weigh slice samples by loss: True
Val split: 0.2
Spurious train split: 0.2
Subsample groups: False
Train method: sc
Max robust acc: -1
Max robust epoch: -1
Max robust group acc: (None, None)
Root dir: ./datasets/data/CelebA/
Target name: Blond_Hair
Confounder names: ['Male']
Image mean: 0.449
Image std: 0.226
Augment data: False
Task: celebA
Num classes: 2
Experiment configs: config
Experiment name: cnc-celebA-sw=re-na=32-np=32-nn=32-nne=0-tsr=1-t=0.05-bf=None-cw=0.5-sud=0-me=5-bst=128-o=sgd-lr=0.0001-mo=0.9-wd=0.1-wdc=0.1-spur-me=1-bst=32-lr=0.001-mo=0.9-wd=0.0005-sts=0.2-s=36-r=0
Mi resampled: None

Loading checkpoints for train split:
[-1 -1 -1 ... -1 -1 -1]
<class 'numpy.ndarray'>
[0 1 2 3] [71629 66874 22880  1387]
Loading checkpoints for val split:
[-1 -1 -1 ... -1  1 -1]
<class 'numpy.ndarray'>
[0 1 2 3] [8535 8276 2874  182]
Loading checkpoints for test split:
[-1 -1 -1 ... -1 -1  1]
<class 'numpy.ndarray'>
[0 1 2 3] [9767 7535 2480  180]
Train dataset:
    Blond_Hair = 0, Male = 0 : n = 71629
    Blond_Hair = 0, Male = 1 : n = 66874
    Blond_Hair = 1, Male = 0 : n = 22880
    Blond_Hair = 1, Male = 1 : n = 1387
Val dataset:
    Blond_Hair = 0, Male = 0 : n = 8535
    Blond_Hair = 0, Male = 1 : n = 8276
    Blond_Hair = 1, Male = 0 : n = 2874
    Blond_Hair = 1, Male = 1 : n = 182
Test dataset:
    Blond_Hair = 0, Male = 0 : n = 9767
    Blond_Hair = 0, Male = 1 : n = 7535
    Blond_Hair = 1, Male = 0 : n = 2480
    Blond_Hair = 1, Male = 1 : n = 180
Pretrained model loaded from 
Epoch:   1 | Train Loss: 0.000 | Train Acc: 85.086 | Val Loss: 0.003 | Val Acc: 84.618
Training:
Accuracies by groups:
0, 0  acc: 71625 / 71629 =  99.994
0, 1  acc: 66868 / 66874 =  99.991
1, 0  acc:     2 / 22880 =   0.009
1, 1  acc:     0 /  1387 =   0.000
--------------------------------------
Average acc: 138495 / 162770 =  85.086
Robust  acc:     0 /  1387 =   0.000
--------------------------------------
Validating:
Accuracies by groups:
0, 0  acc:  8535 /  8535 = 100.000
0, 1  acc:  8276 /  8276 = 100.000
1, 0  acc:     0 /  2874 =   0.000
1, 1  acc:     0 /   182 =   0.000
------------------------------------
Average acc: 16811 / 19867 =  84.618
Robust  acc:     0 /  2874 =   0.000
------------------------------------
Save biased model at epoch 0
replace: True
Checkpoint saved at ./model/celebA/config/stage_one_erm_model_b_epoch0_seed36.pt
New max average-worst acc gap: 84.61770775658127
bias model - Saving best checkpoint at epoch 0
replace: True
Checkpoint saved at ./model/celebA/config/stage_one_erm_model_b_worst_avg_gap_best_epoch0_seed36.pt
-------------------------------------------
Avg Test Loss: 0.003 | Avg Test Acc: 86.675
Robust Acc: 0.000 | Best Acc: 100.000
-------------------------------------
Training, Epoch 0:
Accuracies by groups:
0, 0  acc:  9767 /  9767 = 100.000
0, 1  acc:  7535 /  7535 = 100.000
1, 0  acc:     0 /  2480 =   0.000
1, 1  acc:     0 /   180 =   0.000
------------------------------------
Average acc: 17302 / 19962 =  86.675
Robust  acc:     0 /  2480 =   0.000
------------------------------------
Accuracies by groups:
0, 0  acc:  9767 /  9767 = 100.000
0, 1  acc:  7535 /  7535 = 100.000
1, 0  acc:     0 /  2480 =   0.000
1, 1  acc:     0 /   180 =   0.000
------------------------------------
Average acc: 17302 / 19962 =  86.675
Robust  acc:     0 /  2480 =   0.000
------------------------------------
Testing:
Accuracies by groups:
0, 0  acc:  9767 /  9767 = 100.000
0, 1  acc:  7535 /  7535 = 100.000
1, 0  acc:     0 /  2480 =   0.000
1, 1  acc:     0 /   180 =   0.000
------------------------------------
Average acc: 17302 / 19962 =  86.675
Robust  acc:     0 /  2480 =   0.000
------------------------------------
Epoch:   2 | Train Loss: 0.001 | Train Acc: 87.159 | Val Loss: 0.002 | Val Acc: 91.695
Training:
Accuracies by groups:
0, 0  acc: 71333 / 71629 =  99.587
0, 1  acc: 66869 / 66874 =  99.993
1, 0  acc:  3651 / 22880 =  15.957
1, 1  acc:    16 /  1387 =   1.154
--------------------------------------
Average acc: 141869 / 162770 =  87.159
Robust  acc:    16 /  1387 =   1.154
--------------------------------------
Validating:
Accuracies by groups:
0, 0  acc:  8299 /  8535 =  97.235
0, 1  acc:  8275 /  8276 =  99.988
1, 0  acc:  1636 /  2874 =  56.924
1, 1  acc:     7 /   182 =   3.846
------------------------------------
Average acc: 18217 / 19867 =  91.695
Robust  acc:     7 /   182 =   3.846
------------------------------------
Save biased model at epoch 1
replace: True
Checkpoint saved at ./model/celebA/config/stage_one_erm_model_b_epoch1_seed36.pt
New max average-worst acc gap: 87.8486163758223
bias model - Saving best checkpoint at epoch 1
replace: True
Checkpoint saved at ./model/celebA/config/stage_one_erm_model_b_worst_avg_gap_best_epoch1_seed36.pt
-------------------------------------------
Avg Test Loss: 0.001 | Avg Test Acc: 92.330
Robust Acc: 6.111 | Best Acc: 100.000
-------------------------------------
Training, Epoch 1:
Accuracies by groups:
0, 0  acc:  9600 /  9767 =  98.290
0, 1  acc:  7535 /  7535 = 100.000
1, 0  acc:  1285 /  2480 =  51.815
1, 1  acc:    11 /   180 =   6.111
------------------------------------
Average acc: 18431 / 19962 =  92.330
Robust  acc:    11 /   180 =   6.111
------------------------------------
Accuracies by groups:
0, 0  acc:  9600 /  9767 =  98.290
0, 1  acc:  7535 /  7535 = 100.000
1, 0  acc:  1285 /  2480 =  51.815
1, 1  acc:    11 /   180 =   6.111
------------------------------------
Average acc: 18431 / 19962 =  92.330
Robust  acc:    11 /   180 =   6.111
------------------------------------
Testing:
Accuracies by groups:
0, 0  acc:  9600 /  9767 =  98.290
0, 1  acc:  7535 /  7535 = 100.000
1, 0  acc:  1285 /  2480 =  51.815
1, 1  acc:    11 /   180 =   6.111
------------------------------------
Average acc: 18431 / 19962 =  92.330
Robust  acc:    11 /   180 =   6.111
------------------------------------
Epoch:   3 | Train Loss: 0.000 | Train Acc: 93.158 | Val Loss: 0.001 | Val Acc: 93.909
Training:
Accuracies by groups:
0, 0  acc: 69339 / 71629 =  96.803
0, 1  acc: 66741 / 66874 =  99.801
1, 0  acc: 15378 / 22880 =  67.212
1, 1  acc:   176 /  1387 =  12.689
--------------------------------------
Average acc: 151634 / 162770 =  93.158
Robust  acc:   176 /  1387 =  12.689
--------------------------------------
Validating:
Accuracies by groups:
0, 0  acc:  8197 /  8535 =  96.040
0, 1  acc:  8258 /  8276 =  99.783
1, 0  acc:  2176 /  2874 =  75.713
1, 1  acc:    26 /   182 =  14.286
------------------------------------
Average acc: 18657 / 19867 =  93.909
Robust  acc:    26 /   182 =  14.286
------------------------------------
Save biased model at epoch 2
replace: True
Checkpoint saved at ./model/celebA/config/stage_one_erm_model_b_epoch2_seed36.pt
-------------------------------------------
Avg Test Loss: 0.001 | Avg Test Acc: 94.124
Robust Acc: 17.778 | Best Acc: 99.907
-------------------------------------
Training, Epoch 2:
Accuracies by groups:
0, 0  acc:  9489 /  9767 =  97.154
0, 1  acc:  7528 /  7535 =  99.907
1, 0  acc:  1740 /  2480 =  70.161
1, 1  acc:    32 /   180 =  17.778
------------------------------------
Average acc: 18789 / 19962 =  94.124
Robust  acc:    32 /   180 =  17.778
------------------------------------
Accuracies by groups:
0, 0  acc:  9489 /  9767 =  97.154
0, 1  acc:  7528 /  7535 =  99.907
1, 0  acc:  1740 /  2480 =  70.161
1, 1  acc:    32 /   180 =  17.778
------------------------------------
Average acc: 18789 / 19962 =  94.124
Robust  acc:    32 /   180 =  17.778
------------------------------------
Testing:
Accuracies by groups:
0, 0  acc:  9489 /  9767 =  97.154
0, 1  acc:  7528 /  7535 =  99.907
1, 0  acc:  1740 /  2480 =  70.161
1, 1  acc:    32 /   180 =  17.778
------------------------------------
Average acc: 18789 / 19962 =  94.124
Robust  acc:    32 /   180 =  17.778
------------------------------------
Epoch:   4 | Train Loss: 0.000 | Train Acc: 94.205 | Val Loss: 0.001 | Val Acc: 94.488
Training:
Accuracies by groups:
0, 0  acc: 68842 / 71629 =  96.109
0, 1  acc: 66604 / 66874 =  99.596
1, 0  acc: 17572 / 22880 =  76.801
1, 1  acc:   319 /  1387 =  22.999
--------------------------------------
Average acc: 153337 / 162770 =  94.205
Robust  acc:   319 /  1387 =  22.999
--------------------------------------
Validating:
Accuracies by groups:
0, 0  acc:  8179 /  8535 =  95.829
0, 1  acc:  8258 /  8276 =  99.783
1, 0  acc:  2301 /  2874 =  80.063
1, 1  acc:    34 /   182 =  18.681
------------------------------------
Average acc: 18772 / 19867 =  94.488
Robust  acc:    34 /   182 =  18.681
------------------------------------
Save biased model at epoch 3
replace: True
Checkpoint saved at ./model/celebA/config/stage_one_erm_model_b_epoch3_seed36.pt
-------------------------------------------
Avg Test Loss: 0.001 | Avg Test Acc: 94.835
Robust Acc: 22.222 | Best Acc: 99.814
-------------------------------------
Training, Epoch 3:
Accuracies by groups:
0, 0  acc:  9476 /  9767 =  97.021
0, 1  acc:  7521 /  7535 =  99.814
1, 0  acc:  1894 /  2480 =  76.371
1, 1  acc:    40 /   180 =  22.222
------------------------------------
Average acc: 18931 / 19962 =  94.835
Robust  acc:    40 /   180 =  22.222
------------------------------------
Accuracies by groups:
0, 0  acc:  9476 /  9767 =  97.021
0, 1  acc:  7521 /  7535 =  99.814
1, 0  acc:  1894 /  2480 =  76.371
1, 1  acc:    40 /   180 =  22.222
------------------------------------
Average acc: 18931 / 19962 =  94.835
Robust  acc:    40 /   180 =  22.222
------------------------------------
Testing:
Accuracies by groups:
0, 0  acc:  9476 /  9767 =  97.021
0, 1  acc:  7521 /  7535 =  99.814
1, 0  acc:  1894 /  2480 =  76.371
1, 1  acc:    40 /   180 =  22.222
------------------------------------
Average acc: 18931 / 19962 =  94.835
Robust  acc:    40 /   180 =  22.222
------------------------------------
Epoch:   5 | Train Loss: 0.000 | Train Acc: 94.597 | Val Loss: 0.001 | Val Acc: 94.805
Training:
Accuracies by groups:
0, 0  acc: 68821 / 71629 =  96.080
0, 1  acc: 66572 / 66874 =  99.548
1, 0  acc: 18201 / 22880 =  79.550
1, 1  acc:   381 /  1387 =  27.469
--------------------------------------
Average acc: 153975 / 162770 =  94.597
Robust  acc:   381 /  1387 =  27.469
--------------------------------------
Validating:
Accuracies by groups:
0, 0  acc:  8174 /  8535 =  95.770
0, 1  acc:  8254 /  8276 =  99.734
1, 0  acc:  2365 /  2874 =  82.289
1, 1  acc:    42 /   182 =  23.077
------------------------------------
Average acc: 18835 / 19867 =  94.805
Robust  acc:    42 /   182 =  23.077
------------------------------------
Save biased model at epoch 4
replace: True
Checkpoint saved at ./model/celebA/config/stage_one_erm_model_b_epoch4_seed36.pt
-------------------------------------------
Avg Test Loss: 0.001 | Avg Test Acc: 95.236
Robust Acc: 28.333 | Best Acc: 99.774
-------------------------------------
Training, Epoch 4:
Accuracies by groups:
0, 0  acc:  9464 /  9767 =  96.898
0, 1  acc:  7518 /  7535 =  99.774
1, 0  acc:  1978 /  2480 =  79.758
1, 1  acc:    51 /   180 =  28.333
------------------------------------
Average acc: 19011 / 19962 =  95.236
Robust  acc:    51 /   180 =  28.333
------------------------------------
Accuracies by groups:
0, 0  acc:  9464 /  9767 =  96.898
0, 1  acc:  7518 /  7535 =  99.774
1, 0  acc:  1978 /  2480 =  79.758
1, 1  acc:    51 /   180 =  28.333
------------------------------------
Average acc: 19011 / 19962 =  95.236
Robust  acc:    51 /   180 =  28.333
------------------------------------
Testing:
Accuracies by groups:
0, 0  acc:  9464 /  9767 =  96.898
0, 1  acc:  7518 /  7535 =  99.774
1, 0  acc:  1978 /  2480 =  79.758
1, 1  acc:    51 /   180 =  28.333
------------------------------------
Average acc: 19011 / 19962 =  95.236
Robust  acc:    51 /   180 =  28.333
------------------------------------
replace: True
Checkpoint saved at ./model/celebA/config/bias-end_seed36.pt
training biased model is done
