modelzoo.common.pytorch.run_utils#

Utilities for running Cerebras Pytorch Models

Functions

arg_filter

Checks if a given arg matches the given keyword

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

sideband_eval_all

Temporary support for running eval multiple times via subprocess

sideband_train_eval_all

Temporary support for running train and eval multiple times via subprocess

torchrun

Starts a distributed GPU run using torchrun

update_sideband_mode_arg

Updates sideband arguments to a different mode