cerebras.modelzoo.common.run_utils#

Utilities for running Cerebras Pytorch Models.

Functions

main

Entry point to running pytorch models.

run

Entry point to running pytorch models including CLI argument parsing.

run_with_params

Runs a full end-to-end CS/non-CS workflow for a given model.

set_wsc_log_level

Assert the list of log levels is valid.

torchrun

Starts a distributed GPU run using torchrun.