Skip to content
Snippets Groups Projects
Commit 4f6be619 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Driver] clear ext weights in loader function

parent 87f8f281
No related branches found
No related tags found
No related merge requests found
...@@ -103,8 +103,7 @@ class FINNExampleOverlay(Overlay): ...@@ -103,8 +103,7 @@ class FINNExampleOverlay(Overlay):
Clocks.fclk0_mhz = self.fclk_mhz Clocks.fclk0_mhz = self.fclk_mhz
else: else:
raise ValueError("Supported platforms are zynq-iodma alveo") raise ValueError("Supported platforms are zynq-iodma alveo")
# load any runtime weights # load any external + runtime weights
self.external_weights = []
self.load_external_weights() self.load_external_weights()
self.load_runtime_weights() self.load_runtime_weights()
...@@ -116,6 +115,7 @@ class FINNExampleOverlay(Overlay): ...@@ -116,6 +115,7 @@ class FINNExampleOverlay(Overlay):
weights are one .npy file per layer. weights are one .npy file per layer.
""" """
self.external_weights = []
w_filenames = [] w_filenames = []
if not os.path.isdir(self.runtime_weight_dir): if not os.path.isdir(self.runtime_weight_dir):
return return
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment