ctuning / ck-mlperf

This repository is outdated! Join the open MLPerf workgroup to participate in the development of the next generation of automation workflows for MLPerf benchmarks:
https://bit.ly/mlperf-edu-wg
BSD 3-Clause "New" or "Revised" License
32 stars 23 forks source link

RNN-T CmdGen improvements #59

Open psyhtest opened 3 years ago

psyhtest commented 3 years ago

The RNN-T CmdGen is work-in-progress. We started it for the v0.7 submission round, but eventually did not submit due to a belatedly discovered postprocessing issue. Not surprisingly, it needs more love to get into shape.

Currently, the usage is:

$ ck run cmdgen:benchmark.speech-recognition-loadgen --model=rnnt \
--scenario=singlestream --mode=accuracy \
--sut=aws-g4dn.4xlarge

Future improvements:

psyhtest commented 3 years ago

By the way, the above mentioned postprocessing issue also needs to be fixed but here's a quick workaround (installing the LoadGen Python module to userspace):

$ python -m pip install --user $(find $(ck locate env --tags=mlperf,inference,dividiti.rnnt) -name *.whl)
Processing ./dist/mlperf_loadgen-0.5a0-cp38-cp38-linux_x86_64.whl
Installing collected packages: mlperf-loadgen
Successfully installed mlperf-loadgen-0.5a0