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