My Project
Public Member Functions | List of all members
UnifiedGD_CPU Class Reference

Standard Gradient Descent implementation for CPU. More...

#include <unified_optimization.hpp>

Inheritance diagram for UnifiedGD_CPU:
Inheritance graph
[legend]
Collaboration diagram for UnifiedGD_CPU:
Collaboration graph
[legend]

Public Member Functions

void optimize (NetworkWrapper< CpuBackend > &net, const UnifiedDataset &data, const UnifiedConfig &config) override
 Optimize the network using Gradient Descent on CPU. More...
 
- Public Member Functions inherited from UnifiedOptimizer< CpuBackend >
virtual ~UnifiedOptimizer ()=default
 

Detailed Description

Standard Gradient Descent implementation for CPU.

Member Function Documentation

◆ optimize()

void UnifiedGD_CPU::optimize ( NetworkWrapper< CpuBackend > &  net,
const UnifiedDataset data,
const UnifiedConfig config 
)
inlineoverridevirtual

Optimize the network using Gradient Descent on CPU.

Parameters
netThe network wrapper.
dataThe dataset.
configConfiguration parameters.

Implements UnifiedOptimizer< CpuBackend >.

Here is the call graph for this function:

The documentation for this class was generated from the following file: