parakeet : fix output projection from pre-encode

Author: danbev

src/parakeet.cpp

@@ -1786,20 +1786,21 @@ static struct ggml_cgraph * parakeet_build_graph_conv(parakeet_context & pctx, p
     ggml_set_name(cur, "pre_conv_6_relu");
     ggml_set_output(cur);
 
+    // [freq, time, chan]
+    cur = ggml_permute(ctx0, cur, 0, 2, 1, 3);
+    // [freq, chan, time]
+    cur = ggml_cont(ctx0, cur);
+
     const int n_freq   = cur->ne[0]; // 16
-    const int n_frames = cur->ne[1]; // 188
-    const int n_chan   = cur->ne[2]; // 256
+    const int n_chan   = cur->ne[1]; // 256
+    const int n_frames = cur->ne[2]; // time
 
-    // {n_chan, n_frames, n_freq, batch} -> {d_feat (n_freq * n_chan), n_frames}
-    // {256, 188, 16, 1} -> {4096, 188, 1, 1}
+    // [freq, time, chan, batch] -> [(freq * chan), time]
     cur = ggml_reshape_2d(ctx0, cur, n_freq * n_chan, n_frames);
 
-    // {4096, 188, 1, 1} ->  {1024, 188, 1, 1}
     cur = ggml_mul_mat(ctx0, model.enc_pre_out_w, cur);
     cur = ggml_add(ctx0, cur, model.enc_pre_out_b);
 
-    // {embd, n_time}
-    // {1024, 188, 1, 1}
     ggml_set_name(cur, "embd_conv");
     ggml_set_output(cur);
     pstate.embd_conv = cur;
@@ -2123,6 +2124,8 @@ static bool parakeet_encode_internal(
 
             parakeet_print_tensor_gf("pre_conv_6_relu", sched, gf, 10);
 
+            parakeet_print_tensor_gf(pctx.model.enc_pre_out_w, sched, gf, 10);
+            parakeet_print_tensor_gf(pctx.model.enc_pre_out_b, sched, gf, 10);
             parakeet_print_tensor_gf("embd_conv", sched, gf, 10);
 
             // compare final output: