diff --git a/inference.py b/inference.py index 93428d4421508f9a505e3ca94133191055a0627a..0a44c412192e135d5d5777da53548636e366a78b 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()