epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6961    0.6961    0.6903    0.6903    0.4975    0.555    0.5475    
1    0.6929    0.6929    0.6924    0.6924    0.4962    0.52    0.5686