lbann_library.hpp File Reference

#include "lbann/execution_algorithms/batch_functional_inference_algorithm.hpp"
#include "lbann/models/model.hpp"
#include "lbann/proto/proto_common.hpp"

Include dependency graph for lbann_library.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces
	lbann

Functions
std::unique_ptr< model >	lbann::load_inference_model (lbann_comm *lc, std::string cp_dir, int mbs, std::vector< int > input_dims, std::vector< int > output_dims)
	Loads a trained model from checkpoint for inference only. More...
template<typename DataT , El::Dist CDist, El::Dist RDist, El::DistWrap DistView, El::Device Device>
El::Matrix< int, El::Device::CPU >	lbann::infer (observer_ptr< model > model, El::DistMatrix< DataT, CDist, RDist, DistView, Device > const &samples, size_t mbs)
	Creates execution algorithm and infers on samples using a model. More...
int	lbann::allocate_trainer_resources (lbann_comm *comm)
trainer &	lbann::construct_trainer (lbann_comm *comm, lbann_data::Trainer *pb_trainer, lbann_data::LbannPB &pb)
std::unique_ptr< thread_pool >	lbann::construct_io_thread_pool (lbann_comm *comm, bool serialized_io)
std::unique_ptr< model >	lbann::build_model_from_prototext (int argc, char **argv, const lbann_data::Trainer *pb_trainer, lbann_data::LbannPB &pb, lbann_comm *comm, thread_pool &io_thread_pool, std::vector< std::shared_ptr< callback_base >> &shared_callbacks)
void	lbann::print_lbann_configuration (lbann_comm *comm, int io_threads_per_process, int io_threads_offset)

Variables
const int	lbann::lbann_default_random_seed = 42