model:
  hparams:        
    input_dim: 784
    latent_dim: 784
    distribution: 
      num_classes: 10
      prototypes_per_class: 2
    proto_lr: 0.01
    bb_lr: 0.01