fix minor bug when evaluating

facebookresearch · Aug 12, 2024 · 59ee932 · 59ee932
1 parent 75f6f08
commit 59ee932
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/unibench/evaluator.py b/unibench/evaluator.py
@@ -236,7 +236,7 @@ def evaluate(
                         continue
 
                     if model is None:
-                        load_model(
+                        model = load_model(
                             model_name=model_name,
                             batch_per_gpu=batch_per_gpu,
                             face_blur=face_blur,