epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.693    0.693    0.6939    0.6939    0.5075    0.505    0.4828    
1    0.693    0.693    0.6958    0.6958    0.52    0.485    0.5953