From d5878fff8b8537bed89fe4ee7b8f775986e00b44 Mon Sep 17 00:00:00 2001 From: i-colbert Date: Tue, 6 Feb 2024 10:28:13 -0800 Subject: [PATCH] Fix (examples/a2q): adding input assertion --- src/brevitas_examples/imagenet_classification/a2q/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/brevitas_examples/imagenet_classification/a2q/utils.py b/src/brevitas_examples/imagenet_classification/a2q/utils.py index eaa4c200d..80d5e6389 100644 --- a/src/brevitas_examples/imagenet_classification/a2q/utils.py +++ b/src/brevitas_examples/imagenet_classification/a2q/utils.py @@ -147,6 +147,7 @@ def get_model_by_name( init_from_float_checkpoint: bool = False) -> nn.Module: assert model_name in model_impl, f"Error: {model_name} not implemented." + assert not (pretrained and init_from_float_checkpoint), "Error: pretrained and init_from_float_checkpoint cannot both be true." model: Module = model_impl[model_name]() if init_from_float_checkpoint: