From 7123fa53b73fcba0f80be009de2e83e0d48995f0 Mon Sep 17 00:00:00 2001 From: Yaman Umuroglu Date: Wed, 20 Jul 2022 11:22:46 +0200 Subject: [PATCH] [Hotfix] add missing variable to fix RN50 build --- build/resnet50/build.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/build/resnet50/build.py b/build/resnet50/build.py index 2dae49e..d09c26d 100644 --- 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)