diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 20:47:55 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-05 20:47:55 +0200 |
| commit | 9ae5fa1a88899180f88ddb14d4cef457ceb847e5 (patch) | |
| tree | 4fe2bcd82553c8062eb0908ae6442c123addf55d /text_recognizer/networks/loss/__init__.py | |
| parent | 9e54591b7e342edc93b0bb04809a0f54045c6a15 (diff) | |
Add new training loop with PyTorch Lightning, remove stale files
Diffstat (limited to 'text_recognizer/networks/loss/__init__.py')
| -rw-r--r-- | text_recognizer/networks/loss/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/loss/__init__.py b/text_recognizer/networks/loss/__init__.py index b489264..cb83608 100644 --- a/text_recognizer/networks/loss/__init__.py +++ b/text_recognizer/networks/loss/__init__.py @@ -1,2 +1,2 @@ """Loss module.""" -from .loss import EmbeddingLoss, LabelSmoothingCrossEntropy +from .loss import LabelSmoothingCrossEntropy |