Torchebm > Core > Device_mixin

Contents

Classes

• DeviceMixin - Consistent device management across all modules.

Functions

• normalize_device() - Normalize the device for consistent usage across the library.

API Reference

torchebm.core.device_mixin

Author: Soran Ghaderi Email: soran.gdr.cs@gmail.com This module handles the device management or TorchEBM modules