diff --git a/clip_benchmark/cli.py b/clip_benchmark/cli.py index db2badf..14e3d23 100644 --- 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,