diff --git a/timm/data/parsers/parser_wds.py b/timm/data/parsers/parser_wds.py index 92429009..8e470a38 100644 --- a/timm/data/parsers/parser_wds.py +++ b/timm/data/parsers/parser_wds.py @@ -228,7 +228,7 @@ if wds is not None: seed = pytorch_worker_seed() + epoch else: seed = self.seed + epoch - _logger.info('shuffle', self.seed, epoch, seed) # FIXME temporary + _logger.info(f'shuffle seed: {self.seed}, {seed}, epoch: {epoch}') # FIXME temporary rng = random.Random(seed) return _shuffle(src, self.bufsize, self.initial, rng) @@ -417,11 +417,11 @@ class ParserWds(Parser): ds = self.ds i = 0 - _logger.info('start', i, self.worker_id) # FIXME temporary debug + _logger.info(f'start {i}, {self.worker_id}') # FIXME temporary debug for sample in ds: yield sample[self.image_key], sample[self.target_key] i += 1 - _logger.info('end', i, self.worker_id) # FIXME temporary debug + _logger.info(f'end {i}, {self.worker_id}') # FIXME temporary debug def __len__(self): return math.ceil(max(1, self.repeats) * self.num_samples / self.dist_num_replicas)