Skip to content

Commit

Permalink
add opt parameter
Browse files Browse the repository at this point in the history
  • Loading branch information
benmalef committed Nov 20, 2024
1 parent 101fc30 commit 01ed037
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions GANDLF/utils/pydantic_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ class Parameters(BaseModel):
loss_function: Union[str, Dict]
data_augmentation: dict # TODO: maybe is better to create a class
nested_training: dict # TODO: maybe is better to create a class
optimizer: Union[dict, str] = Field(alias="opt")
optimizer: Union[dict, str]
patch_sampler: Union[dict, str]
patch_size: Union[List[int], int]
clip_mode: Union[str, None]
Expand All @@ -75,7 +75,4 @@ class Parameters(BaseModel):
output_dir: Optional[str] = ""
problem_type: Optional[str] = None
differential_privacy: Optional[dict] = {}

@property
def opt(self):
return self.optimizer
opt: Union[dict, str] = "" #TODO find a better way

0 comments on commit 01ed037

Please sign in to comment.