clipppy.helpers

Module Contents

class clipppy.helpers.scheduled_optimizer_callback

classmethod __call__(cls, i, loss, locs)

static _get_args(i, loss, locs)

class clipppy.helpers.scheduled_optimizer_callback_with_loss

static _get_args(i, loss, locs)

clipppy.helpers.scheduled_optimizer(lr_scheduler_cls: Type[pyro.optim.PyroLRScheduler], optimizer_cls: Type[torch.optim.lr_scheduler._LRScheduler], clip_args=None, **scheduler_kwargs)