(value)
| 29 | |
| 30 | |
| 31 | def set_checkpoint_version(value): |
| 32 | global _CHECKPOINT_VERSION |
| 33 | if _CHECKPOINT_VERSION is not None: |
| 34 | assert _CHECKPOINT_VERSION == value, "checkpoint versions do not match" |
| 35 | _CHECKPOINT_VERSION = value |
| 36 | |
| 37 | |
| 38 | def get_checkpoint_version(): |