class_docstr: Defines how to apply a faster version of the EPR algorithm.
class_docstr_tok: ["Defines","how","to","apply","a","faster","version","of","the","EPR","algorithm","."]
code: def init_training_weights(self, pretrained_weight: tf.Tensor):
      # The probability model is fixed, so only need reparameterization weights.
      self._init_training_weights_reparam(pretrained_weight)
  
imports: import functools
  from typing import Callable, List, Tuple
  import tensorflow
  import tensorflow_compression
  from tensorflow_model_optimization.python.core.common.keras.compression import algorithm
label_desc: Use image node_id 2 for calling the FastEPR obj's underlying member method code with example usage: obj.init_training_weights(pretrained_weight) without return types