peal.adaptors.group_distributionally_robust_optimization.GroupDROConfig¶
- class peal.adaptors.group_distributionally_robust_optimization.GroupDROConfig[source]¶
Bases:
AdaptorConfigConfig template for running the DRO adaptor.
Methods
Attributes
The config template for an adaptor
Parameters for the Gorup_DRO loss computer
A dict containing all variables that could not be given with the current config structure
The path where the model is to be stored.
The config of the predictor to be adapted.
Use paper's datasets.
Resets weights of the model if true.
Reweights groups during training so the training set is balanced.
Use ReduceLROnPlateau scheduler if true.
Sets the seed of the run.
- predictor: PredictorConfig = None¶
The config of the predictor to be adapted.
- replication_dataset: str = None¶
Use paper’s datasets. Only accepts “celebA”, “waterbirds” and None, with None indicating to not use the paper’s datasets, but rather the one specified in predictor’s data config.