epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6956    0.6956    0.6947    0.6947    0.493    0.479    0.5156    
1    0.6937    0.6937    0.6936    0.6936    0.4996    0.489    0.5412