epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.3554    0.3554    0.3274    0.3274    0.8504    0.8718    0.9148    
1    0.3233    0.3233    0.3188    0.3188    0.87    0.8766    0.9186    
2    0.311    0.311    0.3086    0.3086    0.8761    0.8807    0.9244    
3    0.3033    0.3033    0.2912    0.2912    0.8806    0.8837    0.9261    
4    0.3011    0.3011    0.2891    0.2891    0.8815    0.8852    0.9263    
5    0.3009    0.3009    0.3111    0.3111    0.8812    0.8782    0.9202    
6    0.2995    0.2995    0.3177    0.3177    0.882    0.8783    0.924    
7    0.3012    0.3012    0.2969    0.2969    0.8811    0.8827    0.9222    
8    0.2964    0.2964    0.2947    0.2947    0.8838    0.8858    0.9273    
9    0.2958    0.2958    0.2986    0.2986    0.8836    0.8836    0.9238    
10    0.2937    0.2937    0.2985    0.2985    0.885    0.8822    0.9214    
11    0.2931    0.2931    0.3045    0.3045    0.8852    0.8827    0.9229    
12    0.2906    0.2906    0.2898    0.2898    0.8864    0.8846    0.9254    
13    0.2922    0.2922    0.2879    0.2879    0.8851    0.8901    0.9277    
14    0.2903    0.2903    0.2933    0.2933    0.8862    0.8846    0.9247    
15    0.2977    0.2977    0.2909    0.2909    0.8817    0.8872    0.925    
16    0.2848    0.2848    0.3012    0.3012    0.8893    0.8832    0.9214    
17    0.2893    0.2893    0.2842    0.2842    0.8864    0.8899    0.9276    
18    0.2851    0.2851    0.2966    0.2966    0.8887    0.8843    0.9231    
19    0.2935    0.2935    0.2889    0.2889    0.884    0.8857    0.9264    
20    0.2929    0.2929    0.278    0.278    0.8839    0.8919    0.9312    
21    0.2879    0.2879    0.2971    0.2971    0.8867    0.8825    0.9248    
22    0.2874    0.2874    0.2974    0.2974    0.8875    0.8838    0.923    
23    0.2924    0.2924    0.2858    0.2858    0.8841    0.8854    0.9273    
24    0.2867    0.2867    0.2718    0.2718    0.887    0.894    0.9314