[Hotfix] add missing variable to fix RN50 build

Xilinx · Jul 20, 2022 · 7123fa5 · 7123fa5
1 parent 087ad90
commit 7123fa5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/build/resnet50/build.py b/build/resnet50/build.py
@@ -62,7 +62,11 @@
     "step_deployment_package",
 ]
 
-platforms_to_build = ["U250"]
+# which platforms to build the networks for
+zynq_platforms = []
+alveo_platforms = ["U250"]
+platforms_to_build = zynq_platforms + alveo_platforms
+
 model_file = "models/%s_exported.onnx" % model_name
 # create a release dir, used for finn-examples release packaging
 os.makedirs("release", exist_ok=True)