LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
checkpoint_common.hpp File Reference
#include "lbann/models/model.hpp"
#include "lbann/weights/data_type_weights_impl.hpp"
#include <unordered_set>
Include dependency graph for checkpoint_common.hpp:

Go to the source code of this file.

Namespaces

 lbann
 
 lbann::ltfb
 

Functions

static std::string lbann::ltfb::pack (model const &m)
 
static void lbann::ltfb::send_string (lbann_comm const &comm, std::string const &str, int destination_trainer)
 
static std::string lbann::ltfb::recv_string (lbann_comm const &comm, int src_trainer)
 
static void lbann::ltfb::unpack (model &m, std::string const &str)
 
static void lbann::ltfb::restore_model_weights (model &m, std::unordered_map< std::string, std::unique_ptr< weights >> &restore_weights)
 
static std::string lbann::ltfb::sendrecv_string (lbann_comm const &c, std::string const &src, El::Int partner_trainer)
 
template<typename T >
static void lbann::ltfb::exchange (lbann_comm const &c, T &object, El::Int partner_trainer)