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

previous

cerebras.modelzoo.common.utils.run.write_weight_summaries.read_checkpoint

next

cerebras.modelzoo.common.utils.utils.check_and_create_output_dirs