edit

tools/average_models.lua

average_models.lua options:

  • -h [] (default: false)
    This help.
  • -md [] (default: false)
    Dump help in Markdown format.
  • -config (default: '')
    Load options from this file.
  • -save_config (default: '')
    Save options to this file.

Model options

  • -models
    Path to models to average.
  • -output_model (default: final_averaged.t7)
    Path to the averaged model.
  • -force [] (default: false)
    Force output model creation even if the target file exists.
  • Cuda options

    • -gpuid
    (default: 0)
    List of GPU identifiers (1-indexed). CPU is used when set to 0.
  • -fallback_to_cpu [] (default: false)
    If GPU can't be used, rollback on the CPU.
  • -fp16 [] (default: false)
    Use half-precision float on GPU.
  • -no_nccl [] (default: false)
    在并行模式下使得 nccl 不能运用。
  • Logger options

    • -log_file (default: '')
      Output logs to a file under this path instead of stdout.
    • -disable_logs [] (default: false)
      If set, output nothing.
    • -log_level (accepted: DEBUG, INFO, WARNING, ERROR, NOERROR; default: INFO)
      Output logs at this level and above.