Skip to content
Snippets Groups Projects

Support for recurrent networks

Merged Olivier BICHLER requested to merge memorize into dev
24 files
+ 811
46
Compare changes
  • Side-by-side
  • Inline
Files
24
@@ -15,7 +15,7 @@ import aidge_backend_cpu
@@ -15,7 +15,7 @@ import aidge_backend_cpu
from functools import reduce
from functools import reduce
import numpy as np
import numpy as np
class test_recipies(unittest.TestCase):
class test_recipes(unittest.TestCase):
def setUp(self):
def setUp(self):
pass
pass
@@ -33,12 +33,9 @@ class test_recipies(unittest.TestCase):
@@ -33,12 +33,9 @@ class test_recipies(unittest.TestCase):
conv = aidge_core.Conv2D(1, 1, [3, 3], name="Conv0")
conv = aidge_core.Conv2D(1, 1, [3, 3], name="Conv0")
bn = aidge_core.BatchNorm2D(1, name="Add0")
bn = aidge_core.BatchNorm2D(1, name="Add0")
graph_view = aidge_core.sequential([conv, bn])
graph_view = aidge_core.sequential([input_node, conv, bn])
# Add random values to conv and BatchNorm parameters
# Add random values to conv and BatchNorm parameters
input_node.add_child(graph_view)
input_node.get_operator().set_datatype(aidge_core.DataType.Float32)
input_node.get_operator().set_backend("cpu")
graph_view.set_datatype(aidge_core.DataType.Float32)
graph_view.set_datatype(aidge_core.DataType.Float32)
graph_view.set_backend("cpu")
graph_view.set_backend("cpu")
Loading