@Operator(group="train") public final class MergeV2Checkpoints extends PrimitiveOp
result is one logical checkpoint, with one physical metadata file and renamed data files.
Intended for "grouping" multiple checkpoints in a sharded checkpoint setup.
If delete_old_dirs is true, attempts to delete recursively the dirname of each path in the input checkpoint_prefixes. This is useful when those paths are non user-facing temporary locations.
Modifier and Type | Class and Description |
---|---|
static class |
MergeV2Checkpoints.Options
Optional attributes for
MergeV2Checkpoints |
operation
Modifier and Type | Method and Description |
---|---|
static MergeV2Checkpoints |
create(Scope scope,
Operand<String> checkpointPrefixes,
Operand<String> destinationPrefix,
MergeV2Checkpoints.Options... options)
Factory method to create a class wrapping a new MergeV2Checkpoints operation.
|
static MergeV2Checkpoints.Options |
deleteOldDirs(Boolean deleteOldDirs) |
equals, hashCode, op, toString
public static MergeV2Checkpoints create(Scope scope, Operand<String> checkpointPrefixes, Operand<String> destinationPrefix, MergeV2Checkpoints.Options... options)
scope
- current scopecheckpointPrefixes
- prefixes of V2 checkpoints to merge.destinationPrefix
- scalar. The desired final prefix. Allowed to be the same
as one of the checkpoint_prefixes.options
- carries optional attributes valuespublic static MergeV2Checkpoints.Options deleteOldDirs(Boolean deleteOldDirs)
deleteOldDirs
- see above.Copyright © 2022. All rights reserved.