From a7123bc626c7e95eb7449819d24742265ef446fe Mon Sep 17 00:00:00 2001
From: gent <jw02425@surrey.ac.uk>
Date: Sat, 27 Jan 2024 20:42:58 +0000
Subject: [PATCH] Fix indentation in evaluate function
---
vitookit/evaluation/eval_cls.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/vitookit/evaluation/eval_cls.py b/vitookit/evaluation/eval_cls.py
index 8b24b12..7ef3a16 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):
--
GitLab