epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.693    0.693    0.6896    0.6896    0.5088    0.57    0.5532    
1    0.6921    0.6921    0.6932    0.6932    0.52    0.505    0.543