diff --git a/vitookit/evaluation/eval_cls.py b/vitookit/evaluation/eval_cls.py index 8b24b128c7f77db60d92efec51cb06809bcf6906..7ef3a16631a474a7ab2e62f759ebe55603bcf1fb 100644 --- a/vitookit/evaluation/eval_cls.py +++ b/vitookit/evaluation/eval_cls.py @@ -263,7 +263,7 @@ def evaluate(data_loader, model, device,return_preds=False ): if return_preds: return {k: meter.global_avg for k, meter in metric_logger.meters.items()},preds else: - return {k: meter.global_avg for k, meter in metric_logger.meters.items()} + return {k: meter.global_avg for k, meter in metric_logger.meters.items()} def main(args):