Fix run.py

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
2 years ago · 45ed56e2ca
1 changed files with 1 additions and 0 deletions
--- a/benchmark/run.py
+++ b/benchmark/run.py
@ -1,5 +1,6 @@
 import os
 import torch
+from torch import nn
 import numpy
 import onnxruntime