From 45c58b624b2038d9f7036abe00a0129ed292a813 Mon Sep 17 00:00:00 2001 From: Xingjian Shi Date: Mon, 10 Aug 2020 05:06:05 -0700 Subject: [PATCH] Update benchmark_gluonnlp.py --- scripts/benchmarks/benchmark_gluonnlp.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/benchmarks/benchmark_gluonnlp.py b/scripts/benchmarks/benchmark_gluonnlp.py index 7d4687e2b7..7d4c59b446 100644 --- 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 = [