diff --git a/deeprank2/trainer.py b/deeprank2/trainer.py index 0a9a92892..d4a3ead8c 100644 --- a/deeprank2/trainer.py +++ b/deeprank2/trainer.py @@ -749,7 +749,7 @@ def _eval( # pylint: disable=too-many-locals count_predictions += pred.shape[0] sum_of_losses += loss_.detach().item() * pred.shape[0] else: - target_vals = ['None'] * pred.shape[0] + target_vals += ['None'] * pred.shape[0] eval_loss = 'None' # Get the outputs for export