def configs() -> Configs: c = Configs.collect(DummyRegressionDataLoader) c.add( name="input_shape", type=int, default=10, strategy="constant", description="Input shape for data.", ) return c
def configs(): c = Configs.collect(MnistDataset) c.add( name="val_split", type=int, default=5000, strategy="constant", description= "Number samples from train dataset used for validation split.", ) c.add( name="normalize", type=int, default=1, choices=[0, 1], strategy="constant", description="Whether to normalize dataset.", ) return c