From 323fd9c53ab8f62b4c3d07e577ea03ffcaddb945 Mon Sep 17 00:00:00 2001 From: nz11 <nz11@illinois.edu> Date: Sun, 31 Jan 2021 19:02:47 -0600 Subject: [PATCH] Update README.md keras_dump --- hpvm/projects/keras/README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/hpvm/projects/keras/README.md b/hpvm/projects/keras/README.md index 638657a064..4f85276b6f 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 -- GitLab