Skip to content

Commit e8c9e94

Browse files
Internal change
PiperOrigin-RevId: 422864382
1 parent d21a9bb commit e8c9e94

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

official/core/base_trainer.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -370,6 +370,11 @@ def checkpoint(self):
370370
"""Accesses the training checkpoint."""
371371
return self._checkpoint
372372

373+
@property
374+
def checkpoint_exporter(self):
375+
"""Accesses the checkpoint exporter."""
376+
return self._checkpoint_exporter
377+
373378
def train_loop_end(self):
374379
"""See base class."""
375380
self.join()

0 commit comments

Comments
 (0)