cerebras.modelzoo.common.utils.utils#

Functions

check_and_create_output_dirs

configure_compression

Takes in a dictionary of configs and returns the output as a list of compressions

configure_selective_gradient

Takes in a dictionary of selective grad configs and returns the output as a list of SelectiveGrad

format_rate

merge_recursively

Merge new dict into orig dict recursively

save_params

Writes and saves a dictionary to a file in the model_dir.

update_debug_args_with_mem_limits

Update debug args with runconfig memory limits