geowatch.utils.lightning_ext.old_parser_devices module¶
Lightning deprecated its device parser. We should eventually move to their strategy method, but for now we are copying their implementation.
- Augogen:
from pytorch_lightning.utilities import device_parser from lightning_lite.utilities.device_parser import _parse_gpu_ids lib = liberator.Liberator() lib.add_dynamic(_parse_gpu_ids) # lib.expand([‘lightning_lite’]) print(lib.current_sourcecode())
# Broken because liberator fails “as” checks # from lightning_lite.utilities.device_parser import _parse_gpu_ids # from pytorch_lightning.utilities import device_parser # lib = liberator.Liberator() # lib.add_dynamic(device_parser.parse_gpu_ids) # lib.expand([‘lightning_lite.utilities.device_parser’]) # print(lib.current_sourcecode())