class_docstr: Based on <https://arxiv.org/abs/1312.6114>
  Implements VAE loss with a unit Normal anchor.
class_docstr_tok: ["Based","on","<","https",":","\/\/arxiv.org\/abs\/1312.6114",">","Implements","VAE","loss","with","a","unit","Normal","anchor","."]
code: def __init__(self, loss_weight=1.0, min_kl_value=None, name="vae"):
      super().__init__(
          name=name,
          loss_weight=loss_weight,
      )
  
      # minimum value for KL divergence
      if min_kl_value is None:
          self.min_kl_value = min_kl_value
      else:
          self.min_kl_value = float(min_kl_value)
  
imports: import math
  import torch
label_desc: Use image node_id 1 to create a new MegatronVAEHiddenLoss object from inherited base classes: MegatronBaseHiddenLoss with example: obj = MegatronVAEHiddenLoss(loss_weight, min_kl_value, name)