Fix (examples/a2q): adding input assertion

Xilinx · Feb 6, 2024 · d5878ff · d5878ff
1 parent 28c74f7
commit d5878ff
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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: