epoch tot_tloss gw_tloss tot_vloss gw_vloss train_acc valid_acc roc_auc 0 0.6934 0.6934 0.6923 0.6923 0.5065 0.521 0.5744