12 lines
185 B
YAML
12 lines
185 B
YAML
model:
|
|
hparams:
|
|
input_dim: 784
|
|
latent_dim: 784
|
|
distribution:
|
|
num_classes: 10
|
|
prototypes_per_class: 2
|
|
proto_lr: 0.01
|
|
bb_lr: 0.01
|
|
data:
|
|
batch_size: 32
|