Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Dnn.ModelType Module Reference

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.