|
Mila
Deep Neural Network Library
|
Abstract base for language model networks. More...
#include <cstdint>#include <stdexcept>#include <string>import Dnn.TensorDataTypeTraits;import Dnn.TensorDataType;import Compute.DeviceTypeTraits;import Compute.DeviceType;import Dnn.TensorTypes;import Dnn.Tensor;import Dnn.Network;Classes | |
| class | Mila::Dnn::LanguageModelNetwork< TDeviceType, TPrecision > |
| The type erasure boundary between a language model and its transformer. More... | |
| struct | Mila::Dnn::SequenceLogLikelihood |
| Teacher-forced log-likelihood of a token sequence under a model. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
Abstract base for language model networks.