From ed432c2139ca29925bb8b623acd602be1776fe5f Mon Sep 17 00:00:00 2001
From: Ziyang Ma <51037098+ddlBoJack@users.noreply.github.com>
Date: Fri, 20 Oct 2023 20:07:01 +0800
Subject: [PATCH] Update base_audio_only_task.yaml

There is a mismatch between the audio config and the audio checkpoint.
Change the `base_audio_only_task.yaml` file from depth=12, prenet_depth=0 to depth=8, prenet_depth=4 will not affect the training but fix the bug for loading the audio checkpoint.
---
 examples/data2vec/config/v2/base_audio_only_task.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/examples/data2vec/config/v2/base_audio_only_task.yaml b/examples/data2vec/config/v2/base_audio_only_task.yaml
index 65a9ab3e73..e00194d14b 100644
--- a/examples/data2vec/config/v2/base_audio_only_task.yaml
+++ b/examples/data2vec/config/v2/base_audio_only_task.yaml
@@ -66,7 +66,7 @@ model:
   loss_beta: 0
   loss_scale: null
 
-  depth: 12
+  depth: 8
   embed_dim: 768
   clone_batch: 8
 
@@ -91,7 +91,7 @@ model:
       conv_pos_depth: 5
       conv_pos_width: 95
       conv_pos_groups: 16
-      prenet_depth: 0
+      prenet_depth: 4
       mask_prob: 0.5
       mask_prob_adjust: 0.05
       inverse_mask: false