Get an overview of the Cerebras Model Zoo, including model portability, modules, and updated directory paths for enhanced usability.
run
function in the Cerebras Model Zoo and Supported Operations API.
Fig. 7 Cerebras Model Zoo structure
Modules | Description |
---|---|
modelzoo.config | For configuration classes and base-level configuration. |
modelzoo.data | Hosting data processing scripts and loaders for vision and NLP models |
modelzoo.layers | Containing layers and modules for model building and adaptation |
modelzoo.losses | Offering a variety of loss functions for different model training phases |
modelzoo.models | Implementing a range of models across NLP, vision, and multimodal domains |
modelzoo.data_preparation | Providing tools for data preprocessing stages |
modelzoo.tools | Containing utilities for model conversion and configuration |
modelzoo.common | Hosting common utilities across different models |
modelzoo.trainer | Trainer API that facilitates training and validating Model Zoo models |
Old Path (relative to $MODELZOO) | New Path (relative to $MODELZOO) |
---|---|
common/pytorch/* common/run_utils/* common/model_utils/* | common/ |
common/pytorch/layers | layers/ |
common/pytorch/input common/input | data/ |
models/nlp/* vision/pytorch/* multimodal/pytorch/* | models/nlp/* models/vision/* models/multimodal/* |
data_preparation/* | data_preparation/ |
common/pytorch/model_utils/ | tools/ |
fc_mnist/ | fc_mnist/ |