cerebras.modelzoo.common.utils.utils.save_params#

cerebras.modelzoo.common.utils.utils.save_params(params, model_dir, fname='params.yaml')[source]#

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

Parameters

  • params (dict) – dict we want to write to a file in model_dir

  • model_dir (string) – Directory we want to write to

  • fname (string) – Name of file in model_dir we want to save to.

previous

cerebras.modelzoo.common.utils.utils.format_rate

next

cerebras.modelzoo.common.utils.utils.update_debug_args_with_mem_limits