memory
Functions
Utilities related to memory.
-
lightning.pytorch.utilities.memory.garbage_collection_cuda()[source]
Garbage collection Torch (CUDA) memory.
- Return type:
None
-
lightning.pytorch.utilities.memory.recursive_detach(in_dict, to_cpu=False)[source]
Detach all tensors in in_dict.
May operate recursively if some of the values in in_dict are dictionaries
which contain instances of Tensor. Other types in in_dict are
not affected by this utility function.
- Parameters:
-
- Returns:
Dictionary with detached tensors
- Return type:
out_dict