Skip to content

Commit 387d5dd

Browse files
committed
fix dataset
1 parent 96db246 commit 387d5dd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

efficientdet/dataloader.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -397,8 +397,8 @@ def process_example(self, params, batch_size, images, cls_targets,
397397
def dataset_options(self):
398398
options = tf.data.Options()
399399
options.deterministic = self._debug or not self._is_training
400-
options.optimization.map_parallelization = True
401-
options.optimization.parallel_batch = True
400+
options.experimental_optimization.map_parallelization = True
401+
options.experimental_optimization.parallel_batch = True
402402
return options
403403

404404
def __call__(self, params, input_context=None, batch_size=None):
@@ -429,7 +429,7 @@ def _prefetch_dataset(filename):
429429
return dataset
430430

431431
dataset = dataset.interleave(
432-
_prefetch_dataset, num_parallel_calls=tf.data.AUTOTUNE)
432+
_prefetch_dataset, num_parallel_calls=tf.data.AUTOTUNE, deterministic=bool(seed))
433433
dataset = dataset.with_options(self.dataset_options)
434434
if self._is_training:
435435
dataset = dataset.shuffle(64, seed=seed)

0 commit comments

Comments
 (0)