inference
inference copied to clipboard
main.py: error: argument --profile: expected one argument
root@0824378d443e:/inference/vision/classification_and_detection# ./run_local.sh pytorch ssd-mobilenet cpu
usage: main.py [-h]
[--dataset {imagenet,imagenet_mobilenet,imagenet_pytorch,coco-300,coco-300-pt,coco-1200,coco-1200-onnx,coco-1200-pt,coco-1200-tf}]
--dataset-path DATASET_PATH [--dataset-list DATASET_LIST]
[--data-format {NCHW,NHWC}]
[--profile {defaults,resnet50-tf,resnet50-pytorch,resnet50-onnxruntime,mobilenet-tf,mobilenet-onnxruntime,ssd-mobilenet-tf,ssd-mobilenet-pytorch,ssd-mobilenet-onnxruntime,ssd-resnet34-tf,ssd-resnet34-pytorch,ssd-resnet34-onnxruntime,ssd-resnet34-onnxruntime-tf}]
[--scenario SCENARIO] [--max-batchsize MAX_BATCHSIZE] --model
MODEL [--output OUTPUT] [--inputs INPUTS] [--outputs OUTPUTS]
[--backend BACKEND] [--model-name MODEL_NAME]
[--threads THREADS] [--qps QPS] [--cache CACHE] [--accuracy]
[--find-peak-performance] [--debug] [--mlperf_conf MLPERF_CONF]
[--user_conf USER_CONF] [--time TIME] [--count COUNT]
[--max-latency MAX_LATENCY]
[--samples-per-query SAMPLES_PER_QUERY]
main.py: error: argument --profile: expected one argument
Hello, I have a similar issue with pytorch and ssd-mobilenet. Did you manage to address this? Thanks!
outdated