LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
kfac.hpp File Reference
Include dependency graph for kfac.hpp:

Go to the source code of this file.

Classes

class  lbann::KFAC
 An implementation of the KFAC second-order optimization algorithm. More...
 

Namespaces

 lbann
 

Functions

template<>
std::unique_ptr< lbann::KFAClbann::make< lbann::KFAC > (google::protobuf::Message const &msg)
 Build the KFAC training algorithm from a protobuf message. More...
 

Function Documentation

◆ lbann::make< lbann::KFAC >()

template<>
std::unique_ptr<lbann::KFAC> lbann::make< lbann::KFAC > ( google::protobuf::Message const &  msg)

Build the KFAC training algorithm from a protobuf message.