epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6972    0.6972    0.6933    0.6933    0.4988    0.52    0.5408    
1    0.7024    0.7024    0.6984    0.6984    0.4762    0.485    0.509