Skip to content
Snippets Groups Projects
Commit 796d1933 authored by Houssem ROUIS's avatar Houssem ROUIS
Browse files

fix remove_dropout test

parent 152cd114
No related branches found
No related tags found
2 merge requests!59Improvements and fixes,!47Vit operators
...@@ -22,7 +22,7 @@ class test_recipies(unittest.TestCase): ...@@ -22,7 +22,7 @@ class test_recipies(unittest.TestCase):
def test_remove_dropout(self): def test_remove_dropout(self):
graph_view = aidge_core.sequential([ graph_view = aidge_core.sequential([
aidge_core.GenericOperator("Conv", 1, 0, 1, "Conv0"); aidge_core.GenericOperator("Conv", 1, 0, 1, "Conv0"),
aidge_core.GenericOperator("Dropout", 1, 0, 1, name="Dropout0") aidge_core.GenericOperator("Dropout", 1, 0, 1, name="Dropout0")
]) ])
old_nodes = graph_view.get_nodes() old_nodes = graph_view.get_nodes()
......
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