parser.add_argument("--nfx",
default=False,
action="store_true",
help="Choose numpy features extractor")
parser.add_argument("--device",
type=int,
default=0,
help="Device's id to run test on")
args = parser.parse_args()
tf.config.optimizer.set_experimental_options(
{"auto_mixed_precision": args.mxp})
setup_devices([args.device])
from sasegan.runners.tester import SeganTester
from sasegan.datasets.test_dataset import SeganTestDataset
from tensorflow_asr.configs.config import Config
from sasegan.models.sasegan import Generator
from sasegan.featurizers.speech_featurizer import NumpySpeechFeaturizer, TFSpeechFeaturizer
config = Config(args.config)
speech_featurizer = NumpySpeechFeaturizer(config.speech_config) if args.nfx \
else TFSpeechFeaturizer(config.speech_config)
tf.random.set_seed(0)
assert args.saved
parser.add_argument("--subwords",
type=str,
default=None,
help="Path to file that stores generated subwords")
parser.add_argument("--output_name",
type=str,
default="test",
help="Result filename name prefix")
args = parser.parse_args()
tf.config.optimizer.set_experimental_options(
{"auto_mixed_precision": args.mxp})
setup_devices([args.device], cpu=args.cpu)
from tensorflow_asr.configs.config import Config
from tensorflow_asr.datasets.asr_dataset import ASRTFRecordDataset, ASRSliceDataset
from tensorflow_asr.featurizers.speech_featurizers import TFSpeechFeaturizer
from tensorflow_asr.featurizers.text_featurizers import SubwordFeaturizer
from tensorflow_asr.runners.base_runners import BaseTester
from tensorflow_asr.models.conformer import Conformer
config = Config(args.config, learning=True)
speech_featurizer = TFSpeechFeaturizer(config.speech_config)
if args.subwords and os.path.exists(args.subwords):
print("Loading subwords ...")
text_featurizer = SubwordFeaturizer.load_from_file(config.decoder_config,
args.subwords)
import torch.nn as nn
import soundfile as sf
import torch.nn.functional as F
from tamil_tech.torch.utils import *
from tamil_tech.torch.models import *
import numpy as np
import tensorflow as tf
from tensorflow_asr.utils import setup_environment, setup_devices
setup_environment()
tf.config.optimizer.set_experimental_options({"auto_mixed_precision": False})
setup_devices([0], cpu=False)
from tensorflow_asr.configs.config import Config
from tensorflow_asr.featurizers.speech_featurizers import TFSpeechFeaturizer
from tensorflow_asr.featurizers.text_featurizers import CharFeaturizer, SubwordFeaturizer
from tensorflow_asr.models.conformer import Conformer
class ConformerTamilASR(object):
"""
Conformer S based ASR model
"""
def __init__(self, path='ConformerS.h5'):
# fetch and load the config of the model
config = Config('tamil_tech/configs/conformer_new_config.yml', learning=True)
# load speech and text featurizers