diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-03 00:31:00 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-10-03 00:31:00 +0200 |
| commit | 3a21c29e2eff4378c63717f8920ca3ccbfef013c (patch) | |
| tree | ba46504d7baa8d4fb5bfd473acf99a7a184b330c /text_recognizer/criterions | |
| parent | 75eb34020620584247313926527019471411f6af (diff) | |
Lint files
Diffstat (limited to 'text_recognizer/criterions')
| -rw-r--r-- | text_recognizer/criterions/label_smoothing.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/text_recognizer/criterions/label_smoothing.py b/text_recognizer/criterions/label_smoothing.py index 74ff145..5b3a47e 100644 --- a/text_recognizer/criterions/label_smoothing.py +++ b/text_recognizer/criterions/label_smoothing.py @@ -21,7 +21,8 @@ class LabelSmoothingLoss(nn.Module): self, ignore_index: int = -100, smoothing: float = 0.0, dim: int = -1 ) -> None: super().__init__() - assert 0.0 < smoothing <= 1.0 + if not 0.0 < smoothing < 1.0: + raise ValueError("Smoothing must be between 0.0 and 1.0") self.ignore_index = ignore_index self.confidence = 1.0 - smoothing self.smoothing = smoothing |