epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6928    0.6928    0.692    0.692    0.5103    0.5265    0.6583    
1    0.6926    0.6926    0.6925    0.6925    0.5126    0.49    0.7644    
2    0.6919    0.6919    0.6905    0.6905    0.5163    0.4995    0.8345    
3    0.6853    0.6853    0.6759    0.6759    0.5562    0.5205    0.8417    
4    0.6374    0.6374    0.5578    0.5578    0.687    0.7545    0.8353