Default num_samples to None (#227)

neuralmagic · Jun 21, 2023 · 3e7abe4 · 3e7abe4
1 parent 769a692
commit 3e7abe4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sparsify/cli/run.py b/src/sparsify/cli/run.py
@@ -51,7 +51,7 @@ def one_shot(**kwargs):
     one_shot.one_shot(
         model=Path(kwargs["model"]),
         dataset_dir=Path(kwargs["data"]),
-        num_samples=kwargs["train_samples"] or -1,
+        num_samples=kwargs["train_samples"] or None,
         deploy_dir=Path(kwargs["working_dir"]),
         eval_metric=kwargs["eval_metric"],
         optim_level=kwargs["optim_level"],