Fangjun Kuang d9e7f02225 Use torch.jit.script() for positional encoding

2023-06-06 15:25:11 +08:00

..

asr_datamodule.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

beam_search.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

decode_stream.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

decode.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

decoder.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

encoder_interface.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

export-onnx.py

Use torch.jit.script() for positional encoding

2023-06-06 15:25:11 +08:00

export.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

generate_averaged_model.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

jit_pretrained_streaming.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

jit_pretrained.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

joiner.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

model.py

Fix exporting non-streaming zipformer to ONNX via torch.jit.trace()

2023-06-02 14:21:28 +08:00

onnx_pretrained.py

Fix exporting non-streaming zipformer to ONNX via torch.jit.trace()

2023-06-02 14:21:28 +08:00

optim.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

pretrained.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

profile.py

add flops profiler, support for Zipformer encoder and Conformer encoder (#1093)

2023-05-24 19:10:45 +08:00

scaling_converter.py

Use torch.jit.script() for positional encoding

2023-06-06 15:25:11 +08:00

scaling.py

Fix exporting non-streaming zipformer to ONNX via torch.jit.trace()

2023-06-02 14:21:28 +08:00

streaming_beam_search.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

streaming_decode.py

Add the upgraded Zipformer model (#1058)

2023-05-19 16:47:59 +08:00

subsampling.py

Fix exporting non-streaming zipformer to ONNX via torch.jit.trace()

2023-06-02 14:21:28 +08:00

train.py

Remove cur_batch_idx (#1102)

2023-05-30 14:49:54 +08:00

zipformer.py

Use torch.jit.script() for positional encoding

2023-06-06 15:25:11 +08:00