aboutsummaryrefslogtreecommitdiff
path: root/nerv/tnn/tnn.lua
diff options
context:
space:
mode:
authortxh18 <cloudygooseg@gmail.com>2015-12-06 19:31:38 +0800
committertxh18 <cloudygooseg@gmail.com>2015-12-06 19:31:38 +0800
commit27cf4ef1965ec900bb15b32c2b8de65a65c0cd8b (patch)
treed61a4f3e1ea8f27338db99cac55980d7305eb10c /nerv/tnn/tnn.lua
parent4e3cf34a29afbdb17c42ab7f1efacec52efd0e8b (diff)
some log chang
Diffstat (limited to 'nerv/tnn/tnn.lua')
-rw-r--r--nerv/tnn/tnn.lua2
1 files changed, 1 insertions, 1 deletions
diff --git a/nerv/tnn/tnn.lua b/nerv/tnn/tnn.lua
index beb73ca..cf02123 100644
--- a/nerv/tnn/tnn.lua
+++ b/nerv/tnn/tnn.lua
@@ -178,7 +178,7 @@ function TNN:init(batch_size, chunk_size)
nerv.error("layer %s has a zero dim port", ref_from.layer.id)
end
- print("TNN initing storage", ref_from.layer.id, "->", ref_to.layer.id)
+ nerv.info("TNN initing storage %s->%s", ref_from.layer.id, ref_to.layer.id)
ref_to.inputs_matbak_p[port_to] = self.gconf.cumat_type(batch_size, dim)
self.make_initial_store(ref_from.outputs_m, port_from, dim, batch_size, chunk_size, self.extend_t, self.gconf, ref_to.inputs_m, port_to, time)
ref_from.err_inputs_matbak_p[port_from] = self.gconf.cumat_type(batch_size, dim)