class_docstr: ## Configurations
  
  We use [`MNISTConfigs`](../../experiments/mnist.html#MNISTConfigs) configurations.
class_docstr_tok: ["#","#","Configurations","We","use","[","`","MNISTConfigs","`","]","(","..","\/","..","\/experiments\/mnist.html","#","MNISTConfigs",")","configurations","."]
docstr: ### Initialization
docstr_tok: ["#","#","#","Initialization"]
code: def init(self):
      """
      ### Initialization
      """
      # Set tracker configurations
      tracker.set_scalar("loss.*", True)
      tracker.set_scalar("accuracy.*", True)
      tracker.set_histogram("u.*", True)
      tracker.set_histogram("prob.*", False)
      tracker.set_scalar("annealing_coef.*", False)
      tracker.set_scalar("kl_div_loss.*", False)
  
      #
      self.state_modules = []
  
imports: from typing import Any
  import torch.nn
  import torch.utils.data
  from labml import tracker, experiment
  from labml.configs import option, calculate
  from labml_helpers.module import Module
  from labml_helpers.schedule import Schedule, RelativePiecewise
  from labml_helpers.train_valid import BatchIndex
  from labml_nn.experiments.mnist import MNISTConfigs
  from labml_nn.uncertainty.evidence import KLDivergenceLoss, TrackStatistics, MaximumLikelihoodLoss, CrossEntropyBayesRisk, SquaredErrorBayesRisk
label_desc: Use image node_id 1 for calling the Configs obj's underlying member method code with example usage: obj.init() without return types