diff --git a/casanovo/config.py b/casanovo/config.py index 1f3134ed..21174c61 100644 --- a/casanovo/config.py +++ b/casanovo/config.py @@ -89,8 +89,6 @@ class Config: gradient_clip_val=float, gradient_clip_algorithm=str, precision=str, - early_stopping_patience=int, - resume_training_from=str, mskb_tokenizer=bool, ) diff --git a/casanovo/config.yaml b/casanovo/config.yaml index c36d5606..a5e10a6a 100644 --- a/casanovo/config.yaml +++ b/casanovo/config.yaml @@ -109,10 +109,6 @@ gradient_clip_val: gradient_clip_algorithm: precision: "32-true" # '16-true', '16-mixed', 'bf16-true', 'bf16-mixed', '32-true', '64-true', '64', '32', '16', 'bf16' -# Resume training and early stopping -resume_training_from : #'last', 'best', 'path' -early_stopping_patience: - # Replace I by L in peptide sequences replace_isoleucine_with_leucine: True # Reverse peptide sequences diff --git a/casanovo/denovo/model_runner.py b/casanovo/denovo/model_runner.py index 3e085cfc..7016e59b 100644 --- a/casanovo/denovo/model_runner.py +++ b/casanovo/denovo/model_runner.py @@ -114,24 +114,6 @@ def __init__( ), ] - # Configure early stopping - if config.early_stopping_patience is not None: - self.callbacks.append( - EarlyStopping( - monitor="valid_CELoss", - min_delta=0.00, - patience=self.config.early_stopping_patience, - verbose=True, - check_finite=True, - mode="min", - ) - ) - # Configure learning rate monitor - if config.tb_summarywriter is not None: - self.callbacks.append( - LearningRateMonitor(logging_interval="step", log_momentum=True) - ) - def __enter__(self): """Enter the context manager""" self.tmp_dir = tempfile.TemporaryDirectory()