epoch tot_tloss gw_tloss tot_vloss gw_vloss train_acc valid_acc roc_auc 0 0.6966 0.6966 0.6948 0.6948 0.4965 0.4945 0.5078