add to hyperparameters

This commit is contained in:
shahules786 2022-10-02 18:18:32 +05:30
parent b1e15c7552
commit c11bea6aa0
1 changed files with 3 additions and 3 deletions

View File

@ -33,7 +33,7 @@ def main(config: DictConfig):
monitor="val_loss",
mode=direction,
min_delta=0.0,
patience=10,
patience=parameters.get("EarlyStopping_patience",10),
strict=True,
verbose=False,
)
@ -44,10 +44,10 @@ def main(config: DictConfig):
scheduler = ReduceLROnPlateau(
optimizer=optimizer,
mode=direction,
factor=0.1,
factor=parameters.get("ReduceLr_factor",0.1),
verbose=True,
min_lr=parameters.get("min_lr",1e-6),
patience=3
patience=parameters.get("ReduceLr_patience",3)
)
return {"optimizer":optimizer, "lr_scheduler":scheduler}