epoch tot_tloss gw_tloss tot_vloss gw_vloss train_acc valid_acc roc_auc 0 0.6937 0.6937 0.6932 0.6932 0.5008 0.502 0.478