fix

fix

fix

fix

Author: LiTingyu1997

examples/ECAPA-TDNN/speaker_verification_cosine.py

@@ -1,3 +1,5 @@
+# ECAPA_TDNN in mindspore.
+# Adapted from https://github.com/speechbrain/speechbrain/blob/develop/recipes/VoxCeleb/SpeakerRec/speaker_verification_cosine.py
 """
 Recipe for training a speaker verification system based on cosine distance.
 """

examples/ECAPA-TDNN/train_speaker_embeddings.py

@@ -1,3 +1,5 @@
+# ECAPA_TDNN in mindspore.
+# Adapted from https://github.com/speechbrain/speechbrain/blob/develop/recipes/VoxCeleb/SpeakerRec/train_speaker_embeddings.py
 """
 Recipe for training speaker embeddings using the VoxCeleb Dataset.
 """

examples/ECAPA-TDNN/voxceleb_prepare.py

@@ -1,3 +1,5 @@
+# ECAPA_TDNN in mindspore.
+# Adapted from https://github.com/speechbrain/speechbrain/blob/develop/recipes/VoxCeleb/SpeakerRec/voxceleb_prepare.py
 """
 Data preparation, from mindaudio VoxCeleb recipe.
 """

examples/conformer/asr_model.py

@@ -1,3 +1,5 @@
+# Conformer in mindspore.
+# Adapted from https://github.com/wenet-e2e/wenet/blob/main/wenet/transformer/asr_model.py
 """Definition of ASR model."""
 
 import mindspore

examples/conv_tasnet/data.py

@@ -1,3 +1,5 @@
+# AudioDataLoader in mindspore.
+# Adapted from https://github.com/kaituoxu/Conv-TasNet/blob/master/src/data.py
 """
 Logic:
 1. AudioDataLoader generate a minibatch from AudioDataset, the size of this
@@ -16,14 +18,11 @@
     Each targets's shape is B x C x T
 """
 
-import argparse
 import json
 import math
 import os
 
-import mindspore.dataset as ds
 import numpy as np
-from mindspore import context
 
 import mindaudio.data.io as io
 
@@ -176,27 +175,3 @@ def sort_and_pad(self, batch):
 
         sources_pad = sources_pad.transpose((0, 2, 1))
         return mixtures_pad, ilens, sources_pad
-
-
-if __name__ == "__main__":
-    context.set_context(mode=context.GRAPH_MODE, device_target="Ascend", device_id=4)
-    args = parser.parse_args()
-    print(args)
-    tr_dataset = DatasetGenerator(
-        args.train_dir,
-        args.batch_size,
-        sample_rate=args.sample_rate,
-        segment=args.segment,
-    )
-    dataset = ds.GeneratorDataset(
-        tr_dataset, ["mixture", "lens", "sources"], shuffle=False
-    )
-    dataset = dataset.batch(batch_size=5)
-    iter_per_epoch = dataset.get_dataset_size()
-    print(iter_per_epoch)
-    h = 0
-    for data in dataset.create_dict_iterator():
-        h += 1
-        print(data["mixture"])
-        print(data["lens"])
-        print(data["sources"])

examples/conv_tasnet/eval.py

@@ -1,3 +1,5 @@
+# Evaluation of Conv-TasNet in mindspore.
+# Adapted from https://github.com/kaituoxu/Conv-TasNet/blob/master/src/evaluate.py
 import mindspore
 import mindspore.dataset as ds
 import mindspore.ops as ops

examples/conv_tasnet/preprocess.py

@@ -1,3 +1,5 @@
+# Preprocess of Conv-TasNet in mindspore.
+# Adapted from https://github.com/kaituoxu/Conv-TasNet/blob/master/src/preprocess.py
 """ Convert the relevant information in the audio wav file to a json file """
 
 import argparse

examples/conv_tasnet/train.py

@@ -1,3 +1,5 @@
+# Train of Conv-TasNet in mindspore.
+# Adapted from https://github.com/kaituoxu/Conv-TasNet/blob/master/src/train.py
 import os
 
 import mindspore.dataset as ds

examples/deepspeech2/eval.py

@@ -1,3 +1,5 @@
+# Evaluation of deepspeech2 in mindspore.
+# Adapted from https://github.com/SeanNaren/deepspeech.pytorch/tree/master/deepspeech_pytorch/validation.py
 """
 Eval DeepSpeech2
 """

examples/deepspeech2/train.py

@@ -1,3 +1,5 @@
+# Train of deepspeech2 in mindspore.
+# Adapted from https://github.com/SeanNaren/deepspeech.pytorch/tree/master/deepspeech_pytorch/training.py
 """train_criteo."""
 
 import os