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

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