Skip to content
Snippets Groups Projects

Improvements and fixes

Merged Maxence Naud requested to merge tiling into main
39 files
+ 1273
128
Compare changes
  • Side-by-side
  • Inline
Files
39
@@ -20,6 +20,18 @@ class test_recipies(unittest.TestCase):
@@ -20,6 +20,18 @@ class test_recipies(unittest.TestCase):
def tearDown(self):
def tearDown(self):
pass
pass
 
def test_remove_dropout(self):
 
graph_view = aidge_core.sequential([
 
aidge_core.GenericOperator("Conv", 1, 0, 1, "Conv0"),
 
aidge_core.GenericOperator("Dropout", 1, 0, 1, name="Dropout0")
 
])
 
old_nodes = graph_view.get_nodes()
 
aidge_core.remove_dropout(graph_view)
 
self.assertTrue(len(graph_view.get_nodes()) == len(old_nodes) - 1)
 
self.assertTrue("Dropout0" not in [i.name for i in graph_view.get_nodes()])
 
 
self.assertTrue(all([i in old_nodes for i in graph_view.get_nodes()]))
 
def test_remove_flatten(self):
def test_remove_flatten(self):
graph_view = aidge_core.sequential([
graph_view = aidge_core.sequential([
aidge_core.GenericOperator("Flatten", 1, 0, 1, name="Flatten0"),
aidge_core.GenericOperator("Flatten", 1, 0, 1, name="Flatten0"),
Loading