training_context.h

Go to the documentation of this file.

// SPDX-License-Identifier: AGPL-3.0-only
// Copyright (C) 2026 Artificial Intelligence Techniques, S.L. and Roberto López González
 
#pragma once
 
#include "opennn/network/forward_propagation.h"
#include "opennn/network/back_propagation.h"
 
namespace opennn
{
 
class Loss;
 
struct TrainingContext
{
    TrainingContext(Index batch_size, Loss&, bool inputs_pre_scaled = false,
                    TrainingContext* share_memory_with = nullptr,
                    bool joint_gradient_arena = false);
 
    TrainingContext(const TrainingContext&) = delete;
    TrainingContext& operator=(const TrainingContext&) = delete;
 
    bool shares_memory() const noexcept { return shared_memory; }
 
    ForwardPropagation forward;
    BackPropagation backward;
 
private:
    bool shared_memory = false;
};
 
}