Hannes Kuchelmeister
cleanup to make ready as submodule
c7be723
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