epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6196    0.6196    1.3419    1.3419    0.8173    0.7173    0.931    
1    0.5372    0.5372    0.5759    0.5759    0.8509    0.8442    0.9444    
2    0.5334    0.5334    0.5232    0.5232    0.8528    0.8469    0.9474    
3    0.4925    0.4925    0.8754    0.8754    0.8613    0.8202    0.962    
4    0.3789    0.3789    1.0214    1.0214    0.8838    0.5887    0.8597    
5    0.3574    0.3574    1.444    1.444    0.9078    0.5044    0.5479    
6    0.3473    0.3473    1.2957    1.2957    0.9202    0.5327    0.7189    
7    0.348    0.348    1.2488    1.2488    0.9216    0.5865    0.8212    
8    0.3432    0.3432    1.1574    1.1574    0.9224    0.6369    0.8987    
9    0.3409    0.3409    1.0038    1.0038    0.9224    0.6454    0.843