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

fix removeDropout test

parent 05374c04
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):
def test_remove_dropout(self):
graph_view = aidge_core.sequential([
aidge_core.GenericOperator("Conv", 1, 1, 1, "Conv0");
aidge_core.GenericOperator("Conv", 1, 0, 1, "Conv0");
aidge_core.GenericOperator("Dropout", 1, 1, 1, name="Dropout0")
])
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