epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6943    0.6943    0.6969    0.6969    0.5    0.48    0.4909    
1    0.6951    0.6951    0.6919    0.6919    0.5062    0.525    0.4886