From e3571c1ece62ea0f29e2ce6024c3bb10d99b7a1b Mon Sep 17 00:00:00 2001
From: "Eric.Lee2021" <305141918@qq.com>
Date: Sat, 17 Apr 2021 00:51:37 +0800
Subject: [PATCH] update inference

---
 inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference.py b/inference.py
index 93428d4..0a44c41 100644
--- a/inference.py
+++ b/inference.py
@@ -91,7 +91,7 @@ if __name__ == "__main__":
     elif ops.model == "mobilenetv2":
         model_ = MobileNetV2(num_classes=ops.num_classes)
     elif ops.model == "ReXNetV1":
-        model_ = ReXNetV1(num_classes=ops.num_classes)
+        model_ = ReXNetV1( width_mult=0.9, depth_mult=1.0, num_classes=ops.num_classes)
 
     use_cuda = torch.cuda.is_available()
 
-- 
GitLab