epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6985    0.6985    0.6953    0.6953    0.4909    0.483    0.5043    
1    0.6948    0.6948    0.693    0.693    0.4921    0.5065    0.5519