2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Removed experimental AUTOTUNE

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-15 12:14:06 +02:00
parent d6e3cba225
commit 031a1f9ef0
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/data.py
View File

@ -131,7 +131,7 @@ def load_coco(data_path: str, category: int,
dataset = tf.data.Dataset.zip((path_dataset, label_dataset))
dataset = dataset.apply(tf.data.experimental.shuffle_and_repeat(buffer_size=length_dataset, count=num_epochs))
dataset = dataset.batch(batch_size=batch_size)
dataset = dataset.map(_load_image, num_parallel_calls=tf.data.experimental.AUTOTUNE)
dataset = dataset.map(_load_image)
return dataset