infer.yaml 496 B

12345678910111213
  1. # Inheritance
  2. parent_yaml: "base.yaml"
  3. # Data
  4. meta_file: "metadata_test.csv"
  5. # Infer
  6. n_workers: 16 # Num of workers used in data loader.
  7. batch_size: 1 # Batch size.
  8. log_step: 1 # Step interval in logging on command-line and Tensorboard.
  9. use_fp16: False # Usage of half precision.
  10. use_trt: False # Usage of accelerated inference using TensorRT.
  11. pyprof_enabled: False # Usage of pyprof. Enable it only for profiling use.