From 45ed56e2ca85fa7937c8a30225b0ef04cc222b26 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Wed, 14 Dec 2022 19:29:58 +0800
Subject: [PATCH] Fix run.py

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 benchmark/run.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/benchmark/run.py b/benchmark/run.py
index d702f2e..69ed532 100644
--- a/benchmark/run.py
+++ b/benchmark/run.py
@@ -1,5 +1,6 @@
 import os
 import torch
+from torch import nn
 import numpy
 import onnxruntime