diffusion_models.models.nn.CheckpointFunction

class diffusion_models.models.nn.CheckpointFunction(*args, **kwargs)[source]

Bases: Function

__init__(*args, **kwargs)

Methods

__init__(*args, **kwargs)

backward(ctx, *output_grads)

forward(ctx, run_function, length, *args)