|
Mila
Deep Neural Network Library
|
Build-time execution mode for component buffer allocation policy. More...
#include <cstdint>Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
Enumerations | |
| enum class | Mila::Dnn::RuntimeMode : uint8_t { Inference , Training } |
| Runtime mode governing Model API and Network build policy. More... | |
Build-time execution mode for component buffer allocation policy.
|
exportstrong |
Runtime mode governing Model API and Network build policy.
Immutable after Model construction. Determines which public API methods are valid and how the Network allocates its buffers.
| Mode | Network build shape | Valid Model API |
|---|---|---|
| Inference | { 1, context_len } | generate() |
| Training | { batch, seq_len } | eval(), sample() |