From 40b50728dd35ddb5f4889c43a8fcda7f3a4d1d21 Mon Sep 17 00:00:00 2001 From: Huy Le Nguyen Date: Sun, 5 May 2024 22:24:20 +0700 Subject: [PATCH] chore: add tfrecords buffer size to config --- tensorflow_asr/configs.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow_asr/configs.py b/tensorflow_asr/configs.py index 561d3cb68e..a98dd06978 100644 --- a/tensorflow_asr/configs.py +++ b/tensorflow_asr/configs.py @@ -72,6 +72,7 @@ def __init__(self, config: dict = None): self.data_paths = config.pop("data_paths", None) self.tfrecords_dir: str = config.pop("tfrecords_dir", None) self.tfrecords_shards: int = config.pop("tfrecords_shards", 16) + self.tfrecords_buffer_size: int = config.pop("tfrecords_buffer_size", 32 * 1024 * 1024) self.shuffle: bool = config.pop("shuffle", False) self.cache: bool = config.pop("cache", False) self.drop_remainder: bool = config.pop("drop_remainder", True)