prototorch_models/examples/cli/config.yaml
Alexander Engelsberger b60db3174a LightningCLI Example.
2021-05-21 17:13:15 +02:00

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