From 0c7e6439c38675318d49fafefb4dfa18a1eec989 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Tue, 15 Aug 2023 22:19:51 +0200
Subject: [PATCH] [scripts.predict] Reset default batch-size to 1
---
src/ptbench/scripts/predict.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/ptbench/scripts/predict.py b/src/ptbench/scripts/predict.py
index 0715e423..c13a4f24 100644
--- a/src/ptbench/scripts/predict.py
+++ b/src/ptbench/scripts/predict.py
@@ -71,7 +71,7 @@ logger = setup(__name__.split(".")[0], format="%(levelname)s: %(message)s")
requirements for the network).""",
required=True,
show_default=True,
- default=10,
+ default=1,
type=click.IntRange(min=1),
cls=ResourceOption,
)
--
GitLab