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

Stochastic L-BFGS implementation for CPU. More...

#include <unified_optimization.hpp>

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

Public Member Functions

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

Detailed Description

Stochastic L-BFGS implementation for CPU.

Member Function Documentation

◆ optimize()

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

Optimize the network using Stochastic L-BFGS 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: