← all repositories

vahidk/tfrecord

A Python library providing standalone TFRecord file reading and writing with native PyTorch dataset integration.

901 stars Python Data ToolingML Frameworks
tfrecord
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

The tfrecord library enables efficient reading and writing of TensorFlow TFRecord format files in Python. It provides TFRecordDataset and MultiTFRecordDataset classes that implement PyTorch’s IterableDataset interface, allowing seamless integration with PyTorch DataLoader for batching and multi-worker training. The library supports both compressed and uncompressed TFRecords and includes utilities for creating index files to ensure proper multi-worker data loading.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.