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)