parser.add_argument("--subwords_prefix", type=str, default=None, help="Prefix of 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 tiramisu_asr.configs.user_config import UserConfig from tiramisu_asr.datasets.asr_dataset import ASRTFRecordDataset, ASRSliceDataset from tiramisu_asr.featurizers.speech_featurizers import TFSpeechFeaturizer from tiramisu_asr.featurizers.text_featurizers import SubwordFeaturizer from tiramisu_asr.runners.base_runners import BaseTester from tiramisu_asr.models.conformer import Conformer config = UserConfig(DEFAULT_YAML, args.config, learning=True) speech_featurizer = TFSpeechFeaturizer(config["speech_config"]) if args.subwords_prefix and os.path.exists(f"{args.subwords_prefix}.subwords"): print("Loading subwords ...") text_featurizer = SubwordFeaturizer.load_from_file( config["decoder_config"], args.subwords_prefix)
action="store_true", help="Enable mixed precision") parser.add_argument("--device", type=int, default=0, help="Device's id to run test on") parser.add_argument("--bs", type=int, default=None, help="Batch size") args = parser.parse_args() tf.config.optimizer.set_experimental_options( {"auto_mixed_precision": args.mxp}) setup_devices([args.device]) from tiramisu_asr.featurizers.speech_featurizers import TFSpeechFeaturizer from tiramisu_asr.featurizers.text_featurizers import TextFeaturizer from tiramisu_asr.configs.user_config import UserConfig from tiramisu_asr.datasets.asr_dataset import ASRTFRecordDataset, ASRSliceDataset from model import SelfAttentionDS2 from tiramisu_asr.runners.base_runners import BaseTester from ctc_decoders import Scorer tf.random.set_seed(0) assert args.saved config = UserConfig(DEFAULT_YAML, args.config, learning=True) speech_featurizer = TFSpeechFeaturizer(config["speech_config"]) text_featurizer = TextFeaturizer(config["decoder_config"])