epoch tot_tloss gw_tloss tot_vloss gw_vloss train_acc valid_acc roc_auc 0 0.6982 0.6982 0.6928 0.6928 0.4921 0.5105 0.5769