leaspy.algo.utils.algo_with_device module

class AlgoWithDeviceMixin(settings)

Bases: object

Mixin class containing common attributes & methods for algorithms with a torch device.

Parameters:

settingsAlgorithmSettings

The specifications of the algorithm as a AlgorithmSettings instance.

Attributes:

algorithm_devicestr

Valid torch device (only the type of the device is used)