update

toolbox/torchaudio/models/vad/cnn_vad/modeling_cnn_vad.py

@@ -86,6 +86,7 @@ class CNNVadModel(nn.Module):
                  win_type: str,
                  conv2d_block_param_list: List[dict],
                  encoder_output_size: int,
+                 n_frame: int,
                  min_local_snr_db: float = -15,
                  max_local_snr_db: float = 30,
                  ):
@@ -97,6 +98,7 @@ class CNNVadModel(nn.Module):
         self.win_type = win_type
         self.conv2d_block_param_list = conv2d_block_param_list
         self.encoder_output_size = encoder_output_size
+        self.n_frame = n_frame
         self.min_local_snr_db = min_local_snr_db
         self.max_local_snr_db = max_local_snr_db
 
@@ -237,6 +239,7 @@ class CNNVadPretrainedModel(CNNVadModel):
             win_type=config.win_type,
             conv2d_block_param_list=config.conv2d_block_param_list,
             encoder_output_size=config.encoder_output_size,
+            n_frame=config.n_frame,
             min_local_snr_db=config.min_local_snr_db,
             max_local_snr_db=config.max_local_snr_db,
         )