opennn::Optimizer::TrainingSession Struct Reference

#include <opennn/training/optimizer.h>

Classes
struct	GraphPipeline
struct	TailContext

Public Member Functions
Batch *	fixed_batch () const
bool	has_graph_batches () const
void	disable_cuda_graph_capture ()

Public Attributes
array< GraphPipeline, pipelines_count >	pipelines
TailContext	tail
Buffer	device_metrics {Device::CUDA}
array< device::CudaEvent, 4 >	throttle_events
size_t	throttle_cursor = 0
bool	cuda_graph_capture_allowed = false

Static Public Attributes
static constexpr int	group_size = 8
static constexpr int	slots_count = 2 * group_size
static constexpr int	pipelines_count = 2

Member Function Documentation

disable_cuda_graph_capture()

void opennn::Optimizer::TrainingSession::disable_cuda_graph_capture ( )

inline

fixed_batch()

Batch * opennn::Optimizer::TrainingSession::fixed_batch ( ) const

inline

has_graph_batches()

bool opennn::Optimizer::TrainingSession::has_graph_batches ( ) const

inline

Member Data Documentation

cuda_graph_capture_allowed

bool opennn::Optimizer::TrainingSession::cuda_graph_capture_allowed = false

device_metrics

Buffer opennn::Optimizer::TrainingSession::device_metrics {Device::CUDA}

group_size

int opennn::Optimizer::TrainingSession::group_size = 8

staticconstexpr

pipelines

array<GraphPipeline, pipelines_count> opennn::Optimizer::TrainingSession::pipelines

pipelines_count

int opennn::Optimizer::TrainingSession::pipelines_count = 2

staticconstexpr

slots_count

int opennn::Optimizer::TrainingSession::slots_count = 2 * group_size

staticconstexpr

tail

TailContext opennn::Optimizer::TrainingSession::tail

throttle_cursor

size_t opennn::Optimizer::TrainingSession::throttle_cursor = 0

throttle_events

array<device::CudaEvent, 4> opennn::Optimizer::TrainingSession::throttle_events