Function mindspore::dataset::CreateDatasetCache

Function Documentation

inline std::shared_ptr<DatasetCache> mindspore::dataset::CreateDatasetCache(session_id_type id, uint64_t mem_sz, bool spill, const std::optional<std::string> &hostname = std::nullopt, const std::optional<int32_t> &port = std::nullopt, const std::optional<int32_t> &num_connections = std::nullopt, const std::optional<int32_t> &prefetch_sz = std::nullopt)

Function the create a cache to be attached to a dataset.

Parameters
  • id[in] A user assigned session id for the current pipeline.

  • mem_sz[in] Size of the memory set aside for the row caching (default=0 which means unlimited, note that it might bring in the risk of running out of memory on the machine).

  • spill[in] Spill to disk if out of memory (default=False).

  • hostname[in] optional host name (default=”127.0.0.1”).

  • port[in] optional port (default=50052).

  • num_connections[in] optional number of connections (default=12).

  • prefetch_sz[in] optional prefetch size (default=20).

Returns

Shared pointer to DatasetCache. If error, nullptr is returned.

样例
/* Define a Cache object */
std::shared_ptr<DatasetCache> cache = CreateDatasetCache(233, 0, false, "127.0.0.1", 50053, 1, 1);

/* Define dataset path and MindData object */
std::string folder_path = "/path/to/image_directory";
std::shared_ptr<Dataset> ds = ImageFolder(folder_path, false, nullptr, {}, {}, cache);

/* Create iterator to read dataset */
std::shared_ptr<Iterator> iter = ds->CreateIterator();