Stochastic Gradient Descent implementation for CPU (Optimized with Batch Matrix Ops). More...
#include <unified_optimization.hpp>
Inheritance diagram for UnifiedSGD_CPU:
Collaboration diagram for UnifiedSGD_CPU:
Public Member Functions | |
| void | optimize (NetworkWrapper< CpuBackend > &net, const UnifiedDataset &data, const UnifiedConfig &config) override |
| Optimize the network using Stochastic Gradient Descent on CPU. More... | |
 Public Member Functions inherited from UnifiedOptimizer< CpuBackend > | |
| virtual | ~UnifiedOptimizer ()=default |
Detailed Description
Stochastic Gradient Descent implementation for CPU (Optimized with Batch Matrix Ops).
Member Function Documentation
◆ optimize()
|
inlineoverridevirtual |
Optimize the network using Stochastic Gradient Descent on CPU.
- Parameters
-
net The network wrapper. data The dataset. config Configuration parameters.
Implements UnifiedOptimizer< CpuBackend >.
Here is the call graph for this function:
The documentation for this class was generated from the following file:
