Browse Source

Update nnabla script (#759)

Lutz Roeder 3 years ago
parent
commit
98f9bd9001
2 changed files with 2 additions and 0 deletions
  1. 1 0
      Makefile
  2. 1 0
      tools/nnabla

+ 1 - 0
Makefile

@@ -27,6 +27,7 @@ update: install
 	@./tools/dnn schema
 	@./tools/mnn sync schema
 	@./tools/mslite sync schema metadata
+	@./tools/nnabla sync schema metadata
 	@./tools/onnx sync install schema metadata
 	@./tools/om schema
 	@./tools/paddle sync schema

+ 1 - 0
tools/nnabla

@@ -57,6 +57,7 @@ schema() {
 metadata() {
     echo "nnabla metadata"
     venv
+    ${python} -m pip --quiet install pyyaml mako
     export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python
     if [[ $(grep -U $'\x0D' ./source/nnabla-metadata.json) ]]; then crlf=1; else crlf=; fi
     ${python} ./tools/nnabla-script.py metadata