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 |