modelzoo.common.run_utils.cli_parser#

CLI Utilities

Functions

add_csx_arguments

Injects Cerebras System specific parser arguments.

add_general_arguments

Injects general parser arguments.

add_gpu_arguments

Injects GPU specific parser arguments.

assemble_disallowlist

Determine invalid parameters for the current device type.

discard_params

External utility for determining invalid parameters for the current device type.

get_all_args

Helper for returning all valid params for each device.

get_params

Reads params from file and returns them as a dict.

get_params_from_args

Parse the arguments and get the params dict from the resulting args

get_parser

Returns an ArgumentParser for parsing commandline options.

post_process_params

Removes arguments that are not used by this target device.

read_params_file

Helper for loading params file.

update_defaults

Updates the params dict with global default for a key if a key is not present.

update_params_from_args

Update params in-place with the arguments from args.

update_params_from_file

Update provided params from provided file