tools/average_models.lua
average_models.lua
options:
-h [<boolean>]
(default:false
)
This help.-md [<boolean>]
(default:false
)
Dump help in Markdown format.-config <string>
(default:''
)
Load options from this file.-save_config <string>
(default:''
)
Save options to this file.
Model options¶
-models <table>
Path to models to average.-output_model <string>
(default:final_averaged.t7
)
Path to the averaged model.-force [<boolean>]
(default:false
)
Force output model creation even if the target file exists.
Cuda options¶
-gpuid <table>
(default:0
)
List of GPU identifiers (1-indexed). CPU is used when set to 0.-fallback_to_cpu [<boolean>]
(default:false
)
If GPU can't be used, rollback on the CPU.-fp16 [<boolean>]
(default:false
)
Use half-precision float on GPU.-no_nccl [<boolean>]
(default:false
)
Disable usage of nccl in parallel mode.
Logger options¶
-log_file <string>
(default:''
)
Output logs to a file under this path instead of stdout - if file name ending with json, output structure json.-disable_logs [<boolean>]
(default:false
)
If set, output nothing.-log_level <string>
(accepted:DEBUG
,INFO
,WARNING
,ERROR
,NONE
; default:INFO
)
Output logs at this level and above.