diff options
author | txh18 <cloudygooseg@gmail.com> | 2015-12-06 19:31:38 +0800 |
---|---|---|
committer | txh18 <cloudygooseg@gmail.com> | 2015-12-06 19:31:38 +0800 |
commit | 27cf4ef1965ec900bb15b32c2b8de65a65c0cd8b (patch) | |
tree | d61a4f3e1ea8f27338db99cac55980d7305eb10c /nerv/tnn/tnn.lua | |
parent | 4e3cf34a29afbdb17c42ab7f1efacec52efd0e8b (diff) |
some log chang
Diffstat (limited to 'nerv/tnn/tnn.lua')
-rw-r--r-- | nerv/tnn/tnn.lua | 2 |
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) |