From e52414f8d26bd313aa252f6f74844b9e223fbb30 Mon Sep 17 00:00:00 2001
From: Melih Yilmaz <32707537+melihyilmaz@users.noreply.github.com>
Date: Tue, 12 Dec 2023 23:08:12 +0300
Subject: [PATCH] Minor fix to label smoothing

Add option to change label smoothing for model initialization from a checkpoint file.
---
 casanovo/denovo/model_runner.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/casanovo/denovo/model_runner.py b/casanovo/denovo/model_runner.py
index 697442d1..c7a9cab6 100644
--- a/casanovo/denovo/model_runner.py
+++ b/casanovo/denovo/model_runner.py
@@ -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,