Merge pull request mit-han-lab#1 from mit-han-lab/main

Update load_mcunetv3.py to load other models properley
esl-epfl · Aug 2, 2023 · b8f9f1d · b8f9f1d
2 parents 95a0ae6 + f60efb8
commit b8f9f1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/compilation/convert/load_mcunetv3.py b/compilation/convert/load_mcunetv3.py
@@ -23,7 +23,7 @@
 
 def build_quantized_model(net_name="mbv2-w0.35", num_classes=10):
     load_config_from_file("configs/transfer.yaml")
-    configs["net_config"]["net_name"] = "mbv2-w0.35"
+    configs["net_config"]["net_name"] = net_name # "mbv2-w0.35"
     configs["net_config"]["mcu_head_type"] = "quantized"
 
     subnet = build_mcu_model()