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

previous

modelzoo.common.pytorch.run_gpt_inference.main

next

modelzoo.common.pytorch.run_utils.arg_filter