def test_mlp_profiled(self):
if th.config.mode in ("DebugMode", "DEBUG_MODE"):
pytest.skip("Can't profile in DebugMode")
m = models.Mlp()
profile = th.compile.profiling.ProfileStats(False)
f = th.function(m.inputs, m.outputs, profile=profile)
x_val = self.rng.normal(0, 1, (1000, m.nfeatures))
f(x_val)
self.check(f)
def test_mlp(self):
m = models.Mlp()
f = th.function(m.inputs, m.outputs)
pdf = PyDotFormatter()
graph = pdf(f)
expected = 11
if th.config.mode == "FAST_COMPILE":
expected = 12
assert len(graph.get_nodes()) == expected
nc = self.node_counts(graph)
if th.config.mode == "FAST_COMPILE":
assert nc["apply"] == 6
else:
assert nc["apply"] == 5
assert nc["output"] == 1
def test_mlp(self):
m = models.Mlp()
f = th.function(m.inputs, m.outputs)
self.check(f)