cerebras.modelzoo.common.run_utils

Utilities for running Cerebras Pytorch Models.

Functions

main	Runs a full end-to-end CS/non-CS workflow for a PyTorch model.
run	Entry point to running pytorch models including CLI argument parsing.
torchrun	Starts a distributed GPU run using torchrun.

previous

cerebras.modelzoo.common.run_gpt_inference.setup_logging_excepthook

next

cerebras.modelzoo.common.run_utils.main