Skip to content

Commit 3d771b7

Browse files
committed
cnn: Log input dimensions
1 parent 05c25cb commit 3d771b7

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/tinymaix_cnn/mod_cnn.c

+3-2
Original file line numberDiff line numberDiff line change
@@ -129,8 +129,9 @@ static mp_obj_t mod_cnn_new(mp_obj_t model_data_obj) {
129129
memcpy((void*)(o->out_dims), (void*)(&(outs[0])), sizeof(uint16_t)*4);
130130

131131
#if DEBUG
132-
mp_printf(&mp_plat_print, "cnn-new-done outs=%d out.dims=(%d,%d,%d,%d) \n",
133-
outputs, o->out_dims[0], o->out_dims[1], o->out_dims[2], o->out_dims[3]);
132+
mp_printf(&mp_plat_print, "cnn-new-done in.dims=(%d,%d,%d,%d) out.dims=(%d,%d,%d,%d) \n",
133+
o->input.dims, o->input.h, o->input.w, o->input.c,
134+
o->out_dims[0], o->out_dims[1], o->out_dims[2], o->out_dims[3]);
134135
#endif
135136

136137
return MP_OBJ_FROM_PTR(o);

0 commit comments

Comments
 (0)