def test_dump_model():
data_shape = (2, 28)
data = tensor()
data.set_value(np.random.random(data_shape))
mlp = MLP()
pred = mlp(data)
with tempfile.NamedTemporaryFile() as f:
mge.dump(pred, f.name)
def test_dump_model():
data_shape = (2, 28)
data = Tensor(np.random.random(data_shape))
mlp = MLP()
pred = mlp(data)
f = tempfile.NamedTemporaryFile(delete=False)
f_name = f.name
try:
mge.dump(pred, f_name)
finally:
f.close()
os.unlink(f_name)