epoch tot_tloss gw_tloss tot_vloss gw_vloss train_acc valid_acc roc_auc 0 0.6964 0.6964 0.6957 0.6957 0.5049 0.4935 0.5336