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