update inference

e3571c1e · Eric.Lee2021 · ea0e9979 · e3571c1e
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

inference.py inference.py +1 -1

未找到文件。
--- 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()