Update benchmark_gluonnlp.py

liuzh47 · Aug 10, 2020 · 45c58b6 · 45c58b6
1 parent f0ae933
commit 45c58b6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/benchmarks/benchmark_gluonnlp.py b/scripts/benchmarks/benchmark_gluonnlp.py
@@ -1,4 +1,6 @@
+import mxnet as mx
 from benchmark_utils import GluonNLPBackboneBenchmark
+mx.npx.set_np()
 
 
 MODELS = [