From 1420b29ffa01d80e2d80f0cec9a40ac0dd22bf94 Mon Sep 17 00:00:00 2001
From: T5X Team <no-reply@google.com>
Date: Sat, 16 Dec 2023 16:52:12 -0800
Subject: [PATCH] internal

PiperOrigin-RevId: 591577542
---
 t5x/configs/runs/finetune.gin | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/t5x/configs/runs/finetune.gin b/t5x/configs/runs/finetune.gin
index a76d80957..f37944348 100644
--- 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()