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.

previous

cerebras.modelzoo.common.run_gpt_inference.main

next

cerebras.modelzoo.common.run_utils.main