13hannes11
/

master_thesis_models

focus-prediction

Model card Files Files and versions Metrics Training metrics Community

master_thesis_models / configs /datamodule /mnist.yaml

Hannes Kuchelmeister

cleanup to make ready as submodule

c7be723 over 2 years ago

215 Bytes

	_target_: src.datamodules.mnist_datamodule.MNISTDataModule

	data_dir: ${data_dir} # data_dir is specified in config.yaml
	batch_size: 64
	train_val_test_split: [55_000, 5_000, 10_000]
	num_workers: 0
	pin_memory: False