|
Mila
Deep Neural Network Library
|
Enumerations | |
| enum class | Mila::Dnn::ModelType : int { Unknown = 0 , Gpt2 , Llama , Gemma , Mistral , Bert } |
| Canonical list of framework-known model architectures. More... | |
Functions | |
| ModelType | Mila::Dnn::modelTypeFromString (std::string_view s) noexcept |
| Parse a case-insensitive architecture name into a ModelType. | |
| std::string | Mila::Dnn::toString (ModelType t) noexcept |
| Convert a ModelType to its canonical architecture name. | |
Files | |
| file | Mila/Src/Dnn/Core/ModelType.ixx |
| Architecture identity for top-level model networks. | |