Hannes Kuchelmeister
remove old template and use https://github.com/ashleve/lightning-hydra-template instead
d2e7940
raw
history blame
158 Bytes
_target_: src.models.mnist_module.MNISTLitModule
input_size: 784
lin1_size: 256
lin2_size: 256
lin3_size: 256
output_size: 10
lr: 0.001
weight_decay: 0.0005