epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6932    0.6932    0.6941    0.6941    0.5112    0.46    0.4587    
1    0.6931    0.6931    0.6922    0.6922    0.4825    0.545    0.4482