From c639a86c67d449b39a1c08542f5eb06ced84ca84 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 28 Feb 2022 16:28:42 -0800 Subject: [PATCH] Change TFDS default to full re-shuffle (init) each epoch (for now) --- timm/data/parsers/parser_tfds.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/data/parsers/parser_tfds.py b/timm/data/parsers/parser_tfds.py index 132065be..08fa6dbd 100644 --- a/timm/data/parsers/parser_tfds.py +++ b/timm/data/parsers/parser_tfds.py @@ -163,7 +163,7 @@ class ParserTfds(Parser): self.subsplit = None # set when data is distributed across workers using sub-splits self.ds = None # initialized lazily on each dataloader worker process self.init_count = 0 - self.reinit_each_iter = False # self.is_training # FIXME evaluating shuffle across epochs + self.reinit_each_iter = self.is_training # FIXME need to determine if this is necessary def _lazy_init(self): """ Lazily initialize the dataset.