Model.ixx File Reference

#include <cuda_runtime.h>
#include "miniz.h"
#include <vector>
#include <memory>
#include <unordered_map>
#include <string>
#include <iostream>
#include <stdexcept>
#include <type_traits>
#include <functional>
#include <chrono>
import Dnn.ModelCallback;
import Compute.CudaDevice;
import Compute.ComputeDevice;
import Compute.DeviceContext;
import Compute.CudaMemoryResource;
import Compute.CpuDevice;
import Compute.DeviceType;
import Dnn.TensorTraits;
import Data.DataLoader;
import Compute.CpuMemoryResource;
import Dnn.Tensor;
import Dnn.Module;

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...

Namespaces
namespace	Mila
namespace	Mila::Dnn

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.