../pruned_transducer_stateless/test_model.py