10 lines
171 B
YAML
10 lines
171 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
|