antsnormflows.utils.optim

Functions

clear_grad(model)	Set gradients of model parameter to None as this speeds up training,
set_requires_grad(module, flag)	Sets requires_grad flag of all parameters of a torch.nn.module
update_lipschitz(model, n_iterations)