b7a175b7b9
* self-supervised training Signed-off-by: sam1373 <samuelkriman@gmail.com> * test Signed-off-by: sam1373 <samuelkriman@gmail.com> * remove imports Signed-off-by: sam1373 <samuelkriman@gmail.com> * fix Signed-off-by: sam1373 <samuelkriman@gmail.com> * sort imports Signed-off-by: sam1373 <samuelkriman@gmail.com> * fix audio_to_text Signed-off-by: sam1373 <samuelkriman@gmail.com> * manifest handle no text Signed-off-by: sam1373 <samuelkriman@gmail.com> * loss init Signed-off-by: sam1373 <samuelkriman@gmail.com> * style Signed-off-by: sam1373 <samuelkriman@gmail.com> * remove tokenizer from config Signed-off-by: sam1373 <samuelkriman@gmail.com> * config changes Signed-off-by: sam1373 <samuelkriman@gmail.com> * remove hydra import Signed-off-by: sam1373 <samuelkriman@gmail.com> * always spec augment Signed-off-by: sam1373 <samuelkriman@gmail.com> * fixes Signed-off-by: sam1373 <samuelkriman@gmail.com> * copyright Signed-off-by: sam1373 <samuelkriman@gmail.com> * fix cosine sim Signed-off-by: sam1373 <samuelkriman@gmail.com> * fix cosine sim Signed-off-by: sam1373 <samuelkriman@gmail.com> * fix cosine sim Signed-off-by: sam1373 <samuelkriman@gmail.com> * changes based on comments Signed-off-by: sam1373 <samuelkriman@gmail.com> * changes based on comments Signed-off-by: sam1373 <samuelkriman@gmail.com> * configs Signed-off-by: sam1373 <samuelkriman@gmail.com> * name fix Signed-off-by: sam1373 <samuelkriman@gmail.com> * ci config changes Signed-off-by: sam1373 <samuelkriman@gmail.com> * renamed to num_negatives Signed-off-by: sam1373 <samuelkriman@gmail.com> * minor changes Signed-off-by: sam1373 <samuelkriman@gmail.com> * name changes, type annotations Signed-off-by: sam1373 <samuelkriman@gmail.com> Co-authored-by: Yang Zhang <yzhang123@users.noreply.github.com>
49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from nemo.collections.asr.modules.audio_preprocessing import (
|
|
AudioToMelSpectrogramPreprocessor,
|
|
AudioToMFCCPreprocessor,
|
|
CropOrPadSpectrogramAugmentation,
|
|
SpectrogramAugmentation,
|
|
)
|
|
from nemo.collections.asr.modules.beam_search_decoder import BeamSearchDecoderWithLM
|
|
from nemo.collections.asr.modules.conformer_encoder import ConformerEncoder
|
|
from nemo.collections.asr.modules.lstm_decoder import LSTMDecoder
|
|
from nemo.collections.asr.modules.rnnt import RNNTDecoder, RNNTJoint
|
|
|
|
# TODO @blisc: Perhaps refactor instead of import guarding
|
|
try:
|
|
from nemo.collections.asr.modules.conv_asr import (
|
|
ConvASRDecoder,
|
|
ConvASRDecoderClassification,
|
|
ConvASRDecoderReconstruction,
|
|
ConvASREncoder,
|
|
ECAPAEncoder,
|
|
ParallelConvASREncoder,
|
|
SpeakerDecoder,
|
|
)
|
|
except ModuleNotFoundError:
|
|
from nemo.utils.exceptions import CheckInstall
|
|
|
|
# fmt: off
|
|
class ConvASRDecoder(CheckInstall): pass
|
|
class ConvASRDecoderClassification(CheckInstall): pass
|
|
class ConvASREncoder(CheckInstall): pass
|
|
class ECAPAEncoder(CheckInstall): pass
|
|
class ParallelConvASREncoder(CheckInstall): pass
|
|
class SpeakerDecoder(CheckInstall): pass
|
|
class ConvASRDecoderReconstruction(CheckInstall): pass
|
|
# fmt: on
|