13hannes11
/

master_thesis_models

focus-prediction

Model card Files Files and versions Metrics Training metrics Community

master_thesis_models / configs /model /mnist.yaml

Hannes Kuchelmeister

cleanup to make ready as submodule

c7be723 over 2 years ago

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