cerebras.modelzoo.common.run_cstorch_flow.get_cluster_config

cerebras.modelzoo.common.run_cstorch_flow.get_cluster_config(params: dict) → cerebras.appliance.CSConfig.CSConfig

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