Skip to content

Commit 17c4cac

Browse files
committed
Minor fix.
1 parent 086b78c commit 17c4cac

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

efficientnetv2/datasets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,7 @@ def make_source_dataset(self, index, num_hosts):
335335
return tf.data.Dataset.range(1).repeat().map(self._get_null_input)
336336

337337
filenames = tf.io.gfile.glob(
338-
os.path.join(self.cfg.data_dir, self.split_info['files']))
338+
os.path.join(self.data_dir, self.split_info['files']))
339339
filenames = sorted(filenames)[self.split_info['slice']]
340340
for f in filenames[:5]:
341341
logging.info('datafiles: %s', f)

0 commit comments

Comments
 (0)