|
Mila
Deep Neural Network Library
|
Classes | |
| class | Mila::Dnn::Model< TDeviceType, TInput, TOutput > |
| A class representing a neural network model. More... | |
| struct | Mila::Dnn::TrainingConfig |
| Configuration for training a model. More... | |
Typedefs | |
| template<typename TInput = float, typename TOutput = TInput> | |
| using | Mila::Dnn::CpuModel = Model< DeviceType::Cpu, TInput, TOutput > |
| Type alias for CPU-based models with customizable tensor types. | |
| template<typename TInput = float, typename TOutput = TInput> | |
| using | Mila::Dnn::CudaModel = Model< DeviceType::Cuda, TInput, TOutput > |
| Type alias for CUDA-based models with customizable tensor types. | |
Files | |
| file | /home/runner/work/Mila/Mila/Mila/Src/Dnn/Models/Model.ixx |