put model in eval mode before evaluation (#74)

LAION-AI · Jan 30, 2023 · 8d994cf · 8d994cf
1 parent ff19f62
commit 8d994cf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/clip_benchmark/cli.py b/clip_benchmark/cli.py
@@ -173,6 +173,7 @@ def run(args):
             cache_dir=args.model_cache_dir,
             device=args.device
         )
+        model.eval()
         dataset = build_dataset(
             dataset_name=args.dataset, 
             root=dataset_root,