prototorch_models/examples/cli/config.yaml

10 lines
171 B
YAML
Raw Normal View History

2021-05-21 15:10:36 +00:00
model:
hparams:
input_dim: 784
latent_dim: 784
distribution:
num_classes: 10
prototypes_per_class: 2
proto_lr: 0.01
bb_lr: 0.01