diff --git a/unibench/evaluator.py b/unibench/evaluator.py index 29f6cfa..1c5cfd1 100644 --- 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,