noether.data.base.wrappers.timing

Attributes

META_GETITEM_TIME

Classes

TimingWrapper

Wrapper that times __getitem__ calls and returns both the item and the time taken.

Module Contents

noether.data.base.wrappers.timing.META_GETITEM_TIME = '__meta_time_getitem'

class noether.data.base.wrappers.timing.TimingWrapper(dataset)

Bases: torch.utils.data.Dataset

Wrapper that times __getitem__ calls and returns both the item and the time taken.

Parameters:

dataset – The dataset to wrap

dataset