internal

PiperOrigin-RevId: 591578613
google-research · Dec 17, 2023 · 66887b7 · 66887b7
1 parent b9ae6a5
commit 66887b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/t5x/configs/runs/finetune.gin b/t5x/configs/runs/finetune.gin
@@ -140,7 +140,7 @@ utils.SaveCheckpointConfig:
   dtype = 'float32'
   keep = None  # keep all checkpoints
   save_dataset = False  # don't checkpoint dataset state
-
+  restore_dataset = True
 trainer.Trainer:
   num_microbatches = None
   learning_rate_fn = @utils.create_learning_rate_scheduler()