diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:51:59 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-30 23:51:59 +0200 |
| commit | 04876c35a1eeea68794745f1ca113754c0d97757 (patch) | |
| tree | b4664a71c031a9ced6f60c9e72632bad817f11df /text_recognizer/models/base.py | |
| parent | 7268035fb9e57342612a8cc50a1fe04e8841ca2f (diff) | |
black reformatting
Diffstat (limited to 'text_recognizer/models/base.py')
| -rw-r--r-- | text_recognizer/models/base.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/text_recognizer/models/base.py b/text_recognizer/models/base.py index dfb4ca4..caf63c1 100644 --- a/text_recognizer/models/base.py +++ b/text_recognizer/models/base.py @@ -23,12 +23,9 @@ class BaseLitModel(LightningModule): criterion_config: DictConfig = attr.ib(converter=DictConfig) optimizer_config: DictConfig = attr.ib(converter=DictConfig) lr_scheduler_config: DictConfig = attr.ib(converter=DictConfig) - interval: str = attr.ib() monitor: str = attr.ib(default="val/loss") - loss_fn: Type[nn.Module] = attr.ib(init=False) - train_acc: torchmetrics.Accuracy = attr.ib( init=False, default=torchmetrics.Accuracy() ) |