model-io.h File Reference

#include <Eigen/Dense>
#include <vector>
#include <memory>
#include <filesystem>
#include <optional>
#include <future>
#include "fwd.h"
#include "data/types.h"
#include <boost/iterator/iterator_adaptor.hpp>

Go to the source code of this file.

Classes
struct	dismec::io::model::SaveOption
struct	dismec::io::model::WeightFileEntry
	Collect the data about a weight file. More...
class	dismec::io::model::PartialModelIO
	This class is used as an implementation detail to capture the common code of PartialModelSaver and PartialModelLoader. More...
class	dismec::io::model::PartialModelSaver
	Manage saving a model consisting of multiple partial models. More...
class	dismec::io::model::PartialModelLoader
	This class allows loading only a subset of the weights of a large model. More...
struct	dismec::io::model::PartialModelLoader::SubModelRangeSpec

Namespaces
	dismec
	Main namespace in which all types, classes, and functions are defined.
	dismec::io
	dismec::io::model
	namespace for all model-related io functions.

Enumerations
enum class	dismec::io::model::WeightFormat { dismec::io::model::DENSE_TXT = 0 , dismec::io::model::SPARSE_TXT = 1 , dismec::io::model::DENSE_NPY = 2 , dismec::io::model::NULL_FORMAT = 3 }
	Describes the format in which the weight data has been saved. More...

Functions
WeightFormat	dismec::io::model::parse_weights_format (std::string_view name)
	Gets the eighs. More...
const char *	dismec::io::model::to_string (WeightFormat format)
void	dismec::io::model::save_model (const path &target_file, const std::shared_ptr< const Model > &model, SaveOption options)
	Saves a complete model to a file. More...
std::shared_ptr< Model >	dismec::io::model::load_model (path source)