dl-from-scratch / utils /__init__.py
yusiwen's picture
refactor: extract device detection to utils/device.py (CUDA→MPS→CPU)
e9956a9 unverified
Raw
History Blame Contribute Delete
171 Bytes
from .config import load_config, save_config
from .seed import set_seed
from .device import get_device
__all__ = ["load_config", "save_config", "set_seed", "get_device"]