parameters: (self, lr)
docstring: Set the learning rate.
docstring_summary: Set the learning rate.
docstring_tokens: Set the learning rate .
function: def set_learning_rate(self, lr):
          """Set the learning rate."""
          from mitsuba.core import Float
          # Ensure that the JIT compiler does merge 'lr' into the PTX code
          # (this would trigger a recompile every time it is changed)
          self.lr = lr
          self.lr_v = ek.detach(Float(lr, literal=False))
function_tokens: def set_learning_rate ( self , lr ) : from mitsuba . core import Float # Ensure that the JIT compiler does merge 'lr' into the PTX code # (this would trigger a recompile every time it is changed) self . lr = lr self . lr_v = ek . detach ( Float ( lr , literal = False ) )