Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
cli/train.py
View File

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