epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6946    0.6946    0.693    0.693    0.4875    0.5055    0.5718    
1    0.6935    0.6935    0.6929    0.6929    0.5001    0.5005    0.6372    
2    0.693    0.693    0.6927    0.6927    0.5061    0.4925    0.7388    
3    0.693    0.693    0.6917    0.6917    0.514    0.638    0.8216