memory¶
Functions
Garbage collection Torch (CUDA) memory. |
|
|
|
|
|
|
|
|
|
Detach all tensors in in_dict. |
Utilities related to memory.
- lightning.pytorch.utilities.memory.garbage_collection_cuda()[source]¶
Garbage collection Torch (CUDA) memory.
- Return type