epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6913    0.6913    0.6957    0.6957    0.5338    0.5    0.4751    
1    0.6975    0.6975    0.6899    0.6899    0.4925    0.54    0.5289