epoch    tot_tloss    gw_tloss    tot_vloss    gw_vloss    train_acc    valid_acc    roc_auc    
0    0.6955    0.6955    0.6954    0.6954    0.4975    0.49    0.5149    
1    0.6945    0.6945    0.6938    0.6938    0.5038    0.505    0.5229