Initialize checkpoint_manager in train.py.

Resolves lint errors about checkpoint_manager not being initialized. PiperOrigin-RevId: 565162627
google-research · Sep 13, 2023 · b7ff253 · b7ff253
1 parent ea66ec8
commit b7ff253
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/t5x/train.py b/t5x/train.py
@@ -395,6 +395,7 @@ def _verify_matching_vocabs(cfg: utils.DatasetConfig):
 
   # Skip initialization if neither save nor restore is requested.
   train_state = None
+  checkpoint_manager = None
   if valid_restore_cfg or checkpoint_period or checkpoint_steps:
     train_state, checkpoint_manager = (
         utils.create_checkpoint_manager_and_restore(