simulation.utils.machine_learning.data.data_loader module

Summary

Classes:

DataLoader

Wrapper class of Dataset class that performs multi-threaded data loading.

Reference

class DataLoader(dataset, max_dataset_size: int, batch_size: int = 1, num_threads: int = 1, sequential: bool = False)

Bases: object

Wrapper class of Dataset class that performs multi-threaded data loading.

load_data()