Skip to content

Commit

Permalink
rempved to be added functionality (for now)
Browse files Browse the repository at this point in the history
  • Loading branch information
Lilferrit committed Sep 18, 2024
1 parent 281d812 commit 600d0bf
Show file tree
Hide file tree
Showing 3 changed files with 0 additions and 24 deletions.
2 changes: 0 additions & 2 deletions casanovo/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
)

Expand Down
4 changes: 0 additions & 4 deletions casanovo/config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
18 changes: 0 additions & 18 deletions casanovo/denovo/model_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down

0 comments on commit 600d0bf

Please sign in to comment.