trt_multi_engine.yaml 498 B

12345678910111213141516
  1. parent_yaml: 'trt.yaml'
  2. # TRT
  3. trt_multi_engine: True
  4. trt_file_path_list: [
  5. "/fastspeech/preprocessed/v0.2.0/fastspeech.i32.o256.trt",
  6. "/fastspeech/preprocessed/v0.2.0/fastspeech.i64.o512.trt",
  7. "/fastspeech/preprocessed/v0.2.0/fastspeech.i96.o768.trt",
  8. "/fastspeech/preprocessed/v0.2.0/fastspeech.i128.o1024.trt",
  9. ]
  10. trt_max_input_seq_len_list: [32, 64, 96, 128]
  11. trt_max_output_seq_len_list: [256, 512, 768, 1024]
  12. trt_file_path: ""
  13. trt_max_input_seq_len: None
  14. trt_max_output_seq_len: None