LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
lbann::ltfb::CheckpointFile Class Referencefinal

See lbann::callbacks::ltfb::communication_algorithm::checkpoint_file. More...

#include <random_pairwise_exchange.hpp>

Inheritance diagram for lbann::ltfb::CheckpointFile:
[legend]
Collaboration diagram for lbann::ltfb::CheckpointFile:
[legend]

Public Member Functions

 CheckpointFile (std::set< std::string > const &weights_names, std::string const &ckpt_basedir)
 
 CheckpointFile (std::set< std::string > &&weights_names, std::string const &ckpt_basedir)
 
std::unique_ptr< modelget_partner_model (model const &m, El::Int partner_trainer, size_t step) final
 
- Public Member Functions inherited from lbann::Cloneable< CheckpointFile, RandomPairwiseExchange::ExchangeStrategy >
std::unique_ptr< CheckpointFile > clone () const
 Return an exception-safe, memory-safe copy of this object. More...
 

Private Types

using BaseType = Cloneable< CheckpointFile, RandomPairwiseExchange::ExchangeStrategy >
 

Private Attributes

std::string ckpt_basedir_
 

Detailed Description

See lbann::callbacks::ltfb::communication_algorithm::checkpoint_file.

Definition at line 258 of file random_pairwise_exchange.hpp.

Member Typedef Documentation

◆ BaseType

Constructor & Destructor Documentation

◆ CheckpointFile() [1/2]

lbann::ltfb::CheckpointFile::CheckpointFile ( std::set< std::string > const &  weights_names,
std::string const &  ckpt_basedir 
)

◆ CheckpointFile() [2/2]

lbann::ltfb::CheckpointFile::CheckpointFile ( std::set< std::string > &&  weights_names,
std::string const &  ckpt_basedir 
)

Member Function Documentation

◆ get_partner_model()

std::unique_ptr<model> lbann::ltfb::CheckpointFile::get_partner_model ( model const &  m,
El::Int  partner_trainer,
size_t  step 
)
final

Member Data Documentation

◆ ckpt_basedir_

std::string lbann::ltfb::CheckpointFile::ckpt_basedir_
private

Definition at line 273 of file random_pairwise_exchange.hpp.


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