Skip to content
  • API documentation
  • careamics_napari
  • careamics_utils
  • free_memory

free_memory

Utility to free GPU memory.

free_memory(careamist) #

Free memory from CAREamics instance.

Parameters:

Name Type Description Default
careamist CAREamist

CAREamics instance.

 required
Source code in src/careamics_napari/careamics_utils/free_memory.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
def free_memory(careamist: CAREamist) -> None:
    """Free memory from CAREamics instance.

    Parameters
    ----------
    careamist : CAREamist
        CAREamics instance.
    """
    if (
        careamist is not None
        and careamist.trainer is not None
        and careamist.trainer.model is not None
    ):
        careamist.trainer.model.cpu()
        del careamist.trainer.model
        del careamist.trainer
        del careamist

        gc.collect()
        empty_cache()