Skip to content
Snippets Groups Projects
Commit c8d2e55e authored by Georg Streich's avatar Georg Streich
Browse files

Compile conv net

parent 535c45bd
No related branches found
No related tags found
No related merge requests found
......@@ -5,8 +5,6 @@ import numpy as np
import finn.builder.build_dataflow as build
import finn.builder.build_dataflow_config as build_cfg
unsw_nb15_data = np.load("./unsw_nb15_binarized.npz")
inp = np.zeros((1, 3, 32, 32)).astype(np.float32)
out = np.zeros((1)).astype(np.float32)
......@@ -14,10 +12,8 @@ np.save(open("input.npy", "wb"), inp)
np.save(open("expected_output.npy", "wb"), out)
# model_dir = os.environ['FINN_ROOT'] + "/pretrained_models"
# model_file = model_dir + "/cnv-w1a2.onnx"
model_file = '/tmp/finn_dev_streichg/end2end_cnv_w1a1_folded.onnx'
model_dir = os.environ['FINN_ROOT'] + "/pretrained_models"
model_file = model_dir + "/end2end_cnv_w1a1_folded.onnx"
estimates_output_dir = "output_estimates_only"
......
......@@ -16,7 +16,7 @@ class AssignPartitionIDs(Transformation):
self.target_platform,
self.ndevices,
# TODO: Remove this after testing
abs_anchors=[(0, [1]), (1, [4])]
abs_anchors=[(8, [1]), (9, [4])]
)
if floorplans is None:
......
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