epoch tot_tloss gw_tloss tot_vloss gw_vloss train_acc valid_acc roc_auc 0 0.6974 0.6974 0.6933 0.6933 0.4954 0.5035 0.6011