cerebras.modelzoo.config_manager.config_loader#
Utility to load the config from yaml or .py config file
Functions
Utility to convert config object to dict |
|
Convert given dictionary to a config object based on the model name. |
|
Config classes package data related params in a sub dict. |
|
Config classes package optimizer related params in a sub dict. |
|
Config classes package sparsity related params in a sub dict. |