Skip to content

Commit

Permalink
chore: add tfrecords buffer size to config
Browse files Browse the repository at this point in the history
  • Loading branch information
nglehuy committed May 5, 2024
1 parent 316694f commit 40b5072
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tensorflow_asr/configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 40b5072

Please sign in to comment.