diff --git a/hpvm/projects/keras/README.md b/hpvm/projects/keras/README.md index 638657a064a9f97fc448c40ed43c2eb51c5b89d8..4f85276b6f39a22f43e9e9f790232854f49d0e46 100644 --- a/hpvm/projects/keras/README.md +++ b/hpvm/projects/keras/README.md @@ -47,7 +47,7 @@ Activate conda environment (above) before running benchmarks ### Synopsys ``` -python src/${BENCH_NAME}.py [hpvm_reload|keras_reload] [frontend] +python src/${BENCH_NAME}.py [hpvm_reload|keras_reload] [frontend|keras_dump] ``` @@ -59,6 +59,8 @@ python src/${BENCH_NAME}.py [hpvm_reload|keras_reload] [frontend] `frontend`: Invokes the HPVM frontend and dumps weights in directory specified in constructor +`keras_dump`: Dumps keras .h5 format model weights in directory specified in constructor + ### Building New Benchmarks