Skip to content

Commit

Permalink
Minor fix to label smoothing
Browse files Browse the repository at this point in the history
Add option to change label smoothing for model initialization from a checkpoint file.
  • Loading branch information
melihyilmaz authored Dec 12, 2023
1 parent 2aed9e5 commit e52414f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions casanovo/denovo/model_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -242,6 +242,7 @@ def initialize_model(self, train: bool) -> None:
top_match=self.config.top_match,
n_log=self.config.n_log,
tb_summarywriter=self.config.tb_summarywriter,
train_label_smoothing=self.config.train_label_smoothing,
warmup_iters=self.config.warmup_iters,
max_iters=self.config.max_iters,
lr=self.config.learning_rate,
Expand Down

0 comments on commit e52414f

Please sign in to comment.