Skip to content
Snippets Groups Projects

Update BatchNorm and add fuseBatchNorm test

Merged Maxence Naud requested to merge graphview_io_ordering into master
3 files
+ 120
2
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -31,7 +31,7 @@ class test_recipies(unittest.TestCase):
input_node = aidge_core.Producer(input_tensor, "X")
conv = aidge_core.Conv2D(1, 1, [3, 3], name="Conv0")
bn = aidge_core.BatchNorm2D(name="Add0")
bn = aidge_core.BatchNorm2D(1, name="Add0")
graph_view = aidge_core.sequential([conv, bn])
Loading