Mila
Deep Neural Network Library
|
Abstract base class for data loaders used in training and evaluation. More...
#include <type_traits>
#include <cstddef>
import Compute.CpuMemoryResource;
import Compute.CudaPinnedMemoryResource;
import Compute.CudaMemoryResource;
import Compute.MemoryResource;
import Compute.DeviceType;
import Dnn.TensorTraits;
import Dnn.Tensor;
Classes | |
class | Mila::Dnn::Data::DataLoader< TInput, TTarget, TMemoryResource > |
Abstract base class for data loaders used in training and evaluation. More... | |
Namespaces | |
namespace | Mila |
namespace | Mila::Dnn |
namespace | Mila::Dnn::Data |
Abstract base class for data loaders used in training and evaluation.
This module provides a generic data loader interface for efficiently feeding data into neural network models during training and evaluation processes.