JAX DataLoader
Contents:
Installation Guide
Usage Guide
API Reference
Examples
Changelog
JAX DataLoader
Index
Index
_
|
A
|
B
|
C
|
D
|
M
|
N
|
P
|
S
_
__init__() (jax_dataloader.DataLoaderConfig method)
,
[1]
A
augmentation (jax_dataloader.DataLoaderConfig attribute)
auto_batch_size (jax_dataloader.DataLoaderConfig attribute)
B
batch_size (jax_dataloader.DataLoaderConfig attribute)
C
cache_size (jax_dataloader.DataLoaderConfig attribute)
D
DataLoaderConfig (class in jax_dataloader)
M
multi_gpu (jax_dataloader.DataLoaderConfig attribute)
N
num_workers (jax_dataloader.DataLoaderConfig attribute)
P
pinned_memory (jax_dataloader.DataLoaderConfig attribute)
prefetch (jax_dataloader.DataLoaderConfig attribute)
progress_tracking (jax_dataloader.DataLoaderConfig attribute)
S
shuffle (jax_dataloader.DataLoaderConfig attribute)