epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.3977    0.3977    0.3273    0.3273    0.82    0.8686    0.9126    
1    0.3288    0.3288    0.3177    0.3177    0.8647    0.8732    0.9185    
2    0.3129    0.3129    0.3355    0.3355    0.8736    0.8698    0.9156    
3    0.311    0.311    0.3224    0.3224    0.8742    0.8731    0.9147    
4    0.3107    0.3107    0.2956    0.2956    0.8735    0.8832    0.9251    
5    0.306    0.306    0.3117    0.3117    0.8761    0.8745    0.9206    
6    0.3006    0.3006    0.3114    0.3114    0.8792    0.8761    0.9204    
7    0.301    0.301    0.294    0.294    0.8786    0.8846    0.9257    
8    0.2982    0.2982    0.2906    0.2906    0.8806    0.8844    0.9255    
9    0.2997    0.2997    0.3027    0.3027    0.8793    0.88    0.9199    
10    0.2942    0.2942    0.3173    0.3173    0.8828    0.8766    0.9181    
11    0.2955    0.2955    0.3095    0.3095    0.8817    0.8772    0.9214    
12    0.2971    0.2971    0.2842    0.2842    0.8801    0.888    0.9276    
13    0.2882    0.2882    0.2812    0.2812    0.8852    0.8901    0.929    
14    0.2927    0.2927    0.2909    0.2909    0.8831    0.8871    0.9253    
15    0.2923    0.2923    0.3078    0.3078    0.8833    0.8801    0.9218    
16    0.2983    0.2983    0.2861    0.2861    0.8793    0.8888    0.927    
17    0.2915    0.2915    0.2918    0.2918    0.8837    0.8861    0.9249    
18    0.2883    0.2883    0.285    0.285    0.8851    0.8889    0.9262    
19    0.287    0.287    0.2875    0.2875    0.8858    0.8888    0.9268    
20    0.2858    0.2858    0.2849    0.2849    0.8864    0.8894    0.9282    
21    0.2869    0.2869    0.2917    0.2917    0.886    0.8861    0.9251    
22    0.2895    0.2895    0.2941    0.2941    0.8842    0.8873    0.9259    
23    0.2876    0.2876    0.2831    0.2831    0.8851    0.8893    0.927    
24    0.2892    0.2892    0.2863    0.2863    0.885    0.8871    0.9267