cerebras.modelzoo.common.run_cstorch_flow.get_cluster_config#

cerebras.modelzoo.common.run_cstorch_flow.get_cluster_config(params)[source]#

Sets up CS cluster config for the run.

previous

cerebras.modelzoo.common.run_cstorch_flow.compute_params_norm

next

cerebras.modelzoo.common.run_cstorch_flow.get_model_checkpoint