-
Notifications
You must be signed in to change notification settings - Fork 225
Closed
Description
Hi,
I am running RN50 workloads, and I want to set the "--steps=1000" parameter with the scripts of
models/benchmarks/launch_benchmark.py
My command would be:
python models/benchmarks/launch_benchmark.py --in-graph /root/rsn50_frozen_max_min.pb --model-name resnet50 --framework tensorflow --precision int8 --mode inference --batch-size=1 --socket-id 0 --benchmark-only --docker-image tensorflow/tensorflow-estimator:latest-mkl --data-location /root/tensorflow/dataset/TF_Imagenet_FullData --num-inter-threads=1 --num-intra-threads=26 --steps=1000
The "--steps" parameter doesn't work.
And I have to modified the function "add_steps_args()" in the "benchmarks/common/tensorflow/start.sh” script
Metadata
Metadata
Assignees
Labels
No labels