MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / load_snapshot

Method load_snapshot

python/oneflow/one_embedding.py:284–299  ·  view source on GitHub ↗

load snapshot Args: snapshot_name (str): the snapshot_name, snapshot will be load from your_configed_persistent_path For example: .. code-block:: python >>> import oneflow as flow >>> # use embedding create by flow.one_embedding.Mul

(self, snapshot_name)

Source from the content-addressed store, hash-verified

282 self.handler.SaveSnapshot(snapshot_name)
283
284 def load_snapshot(self, snapshot_name):
285 """load snapshot
286
287 Args:
288 snapshot_name (str): the snapshot_name, snapshot will be load from your_configed_persistent_path
289
290 For example:
291
292 .. code-block:: python
293
294 >>> import oneflow as flow
295 >>> # use embedding create by flow.one_embedding.MultiTableEmbedding
296 >>> embedding.load_snapshot("my_snapshot1")
297 >>> # load a snapshot named "my_snapshot1" from your_configed_persistent_path
298 """
299 self.handler.LoadSnapshot(snapshot_name)
300
301 def forward(self, ids, table_ids=None):
302 """Embedding lookup operation

Callers

nothing calls this directly

Calls 1

LoadSnapshotMethod · 0.45

Tested by

no test coverage detected