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