Config
Config
dataclass
#
The options required for a run. This dataclass acts as a structure for the Hydra configs.
For more info, see https://hydra.cc/docs/tutorials/structured_config/schema/
algorithm
instance-attribute
#
algorithm: Any
Configuration for the algorithm (a LightningModule).
It is suggested for this class to accept a datamodule
and network
as arguments. The
instantiated datamodule and network will be passed to the algorithm's constructor.
For more info, see the instantiate_algorithm function.
datamodule
class-attribute
instance-attribute
#
Configuration for the datamodule (dataset + transforms + dataloader creation).
This should normally create a LightningDataModule. See the MNISTDataModule for an example.
trainer
class-attribute
instance-attribute
#
Keyword arguments for the Trainer constructor.
seed
class-attribute
instance-attribute
#
Random seed for reproducibility.
If None, a random seed is generated.