from tensorflow_asr.datasets.asr_dataset import ASRTFRecordDataset, ASRSliceDataset
from tensorflow_asr.featurizers.speech_featurizers import TFSpeechFeaturizer
from tensorflow_asr.featurizers.text_featurizers import CharFeaturizer
from tensorflow_asr.runners.base_runners import BaseTester
from tensorflow_asr.models.jasper import Jasper
tf.random.set_seed(0)
assert args.export
config = Config(args.config, learning=True)
speech_featurizer = TFSpeechFeaturizer(config.speech_config)
text_featurizer = CharFeaturizer(config.decoder_config)
# Build DS2 model
jasper = Jasper(**config.model_config, vocabulary_size=text_featurizer.num_classes)
jasper._build(speech_featurizer.shape)
jasper.load_weights(args.saved, by_name=True)
jasper.summary(line_length=120)
jasper.add_featurizers(speech_featurizer, text_featurizer)
if args.tfrecords:
test_dataset = ASRTFRecordDataset(
data_paths=config.learning_config.dataset_config.test_paths,
tfrecords_dir=config.learning_config.dataset_config.tfrecords_dir,
speech_featurizer=speech_featurizer,
text_featurizer=text_featurizer,
stage="test", shuffle=False
)
else:
test_dataset = ASRSliceDataset(
data_paths=config.learning_config.dataset_config.test_paths,
speech_featurizer=speech_featurizer,
from tensorflow_asr.featurizers.speech_featurizers import TFSpeechFeaturizer
from tensorflow_asr.featurizers.text_featurizers import CharFeaturizer
from tensorflow_asr.runners.base_runners import BaseTester
from tensorflow_asr.models.jasper import Jasper
tf.random.set_seed(0)
#assert args.export
config = Config(args.config)
speech_featurizer = TFSpeechFeaturizer(config.speech_config)
text_featurizer = CharFeaturizer(config.decoder_config)
# Build DS2 model
jasper = Jasper(**config.model_config,
vocabulary_size=text_featurizer.num_classes)
jasper._build(speech_featurizer.shape)
jasper.load_weights(args.saved)
jasper.summary(line_length=120)
jasper.add_featurizers(speech_featurizer, text_featurizer)
if args.tfrecords:
test_dataset = ASRTFRecordDataset(
speech_featurizer=speech_featurizer,
text_featurizer=text_featurizer,
**vars(config.learning_config.test_dataset_config))
else:
test_dataset = ASRSliceDataset(
speech_featurizer=speech_featurizer,
text_featurizer=text_featurizer,
**vars(config.learning_config.test_dataset_config))
ctc_tester = BaseTester(config=config.learning_config.running_config,