convert.yaml 261 B

123456789101112
  1. # SPDX-License-Identifier: Apache-2.0
  2. _target_: inference.converter.run_converter
  3. defaults:
  4. - export: ts-trace
  5. - convert: torchscript
  6. config:
  7. checkpoint: ???
  8. batch_size: 64
  9. precision: fp32
  10. optimize: False
  11. accelerator: none
  12. gpu: 0