pytorch_pfn_extras.profiler.TraceableDataset#

class pytorch_pfn_extras.profiler.TraceableDataset(dataset, tag, tracer=None)#

Bases: Dataset

Utility class to trace a Dataset inside the DataLoader worker threads.

Parameters:

  • dataset (torch.utils.data.Dataset) – dataset where __getitem__ will be traced.

  • tag (str) – Tag will be used to name the events.

  • tracer (Tracer) – Tracer object, optional. If None it defaults to ppe.profile.get_tracer().

Methods

__init__(dataset, tag[, tracer])

__init__(dataset, tag, tracer=None)#
Parameters:

  • dataset (Dataset) –

  • tag (str) –

  • tracer (Optional[Tracer]) –

Return type:

None