cerebras.modelzoo.common.train_and_eval#

Generic run scripts build using the cstorch API.

Functions

train_and_eval

Runs the train and eval workflow built using the cstorch API.

validate_train_and_eval_micro_batch_size

Validate training and evaluation batch_size/micro_batch_size settings.