class_docstr: Reset all weights of some layers.
  
  This class implements the primacy bias paper, in which all weights of the
  last layers of the model are randomly reinitalized.
  
  Attributes:
class_docstr_tok: ["Reset","all","weights","of","some","layers",".","This","class","implements","the","primacy","bias","paper",",","in","which","all","weights","of","the","last","layers","of","the","model","are","randomly","reinitalized",".","Attributes",":"]
code: def is_reset(self, update_step):
      within_reset_interval = (
          update_step >= self.reset_start_step
          and update_step < self.reset_end_step
      )
      return self.is_update_iter(update_step) and within_reset_interval
  
imports: import functools
  import logging
  import flax
  from flax import linen
  import gin
  import jax
  from jax import random
  import jax.numpy
  import optax
label_desc: Use image node_id 1 for calling the LayerReset obj's underlying member method code with example usage: obj.is_reset(update_step) and returns: unknown