MCPcopy Create free account
hub / github.com/Meshcapade/difflocks / __init__

Method __init__

k_diffusion/utils.py:173–181  ·  view source on GitHub ↗
(self, optimizer, inv_gamma=1., power=1., warmup=0., min_lr=0.,
                 last_epoch=-1, verbose=False)

Source from the content-addressed store, hash-verified

171 """
172
173 def __init__(self, optimizer, inv_gamma=1., power=1., warmup=0., min_lr=0.,
174 last_epoch=-1, verbose=False):
175 self.inv_gamma = inv_gamma
176 self.power = power
177 if not 0. <= warmup < 1:
178 raise ValueError('Invalid value for warmup')
179 self.warmup = warmup
180 self.min_lr = min_lr
181 super().__init__(optimizer, last_epoch, verbose)
182
183 def get_lr(self):
184 if not self._get_lr_called_within_step:

Callers

nothing calls this directly

Calls 1

__init__Method · 0.45

Tested by

no test coverage detected