epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.694    0.694    0.6929    0.6929    0.4986    0.5135    0.491    
1    0.6944    0.6944    0.6932    0.6932    0.4915    0.483    0.4895