nncf.common.initialization.dataloader#

Interface for user-defined data usage during the compression algorithm initialization process.

Classes#

NNCFDataLoader

Wraps a custom data source.
class nncf.common.initialization.dataloader.NNCFDataLoader[source]#

Bases: abc.ABC

Wraps a custom data source.

abstract property batch_size: int[source]#

Returns the number of elements return per iteration.

Returns:

A number of elements return per iteration.

Return type:

int

abstract __iter__()[source]#

Creates an iterator for the elements of a custom data source. The returned iterator implements the Python Iterator protocol.

Returns:

An iterator for the elements of a custom data source.