From af7646289edef4ff2d59e70b2bc5b1b63957f6d5 Mon Sep 17 00:00:00 2001 From: Chuang Zhu Date: Wed, 6 Dec 2023 08:55:09 +0000 Subject: [PATCH] fix inferencesample_option --- python/pylibwholegraph/pylibwholegraph/torch/common_options.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/pylibwholegraph/pylibwholegraph/torch/common_options.py b/python/pylibwholegraph/pylibwholegraph/torch/common_options.py index 0999fdfe5..42746add8 100644 --- a/python/pylibwholegraph/pylibwholegraph/torch/common_options.py +++ b/python/pylibwholegraph/pylibwholegraph/torch/common_options.py @@ -132,7 +132,7 @@ def add_common_sampler_options(argparser: ArgumentParser): argparser.add_argument( "-s", "--inferencesample", - type=int, + type=str, dest="inferencesample", default="30", help="inference sample count, -1 is all",