epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6974    0.6974    0.7009    0.7009    0.5012    0.475    0.4866    
1    0.7027    0.7027    0.6936    0.6936    0.47    0.515    0.5249