summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-27 20:27:25 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-27 20:27:25 +0200
commit6f21058f55ab9146673cf0098dc27869aee4090c (patch)
tree5287104e126b0fe65558b094f6a58569655bc03a /text_recognizer
parentd08a9f87a86ae6ed4a1077a3038fc444085d7f6e (diff)
Remove unused imports
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/models/transformer.py5
-rw-r--r--text_recognizer/models/vqvae.py10
2 files changed, 8 insertions, 7 deletions
diff --git a/text_recognizer/models/transformer.py b/text_recognizer/models/transformer.py
index 8dd4db2..bc7e313 100644
--- a/text_recognizer/models/transformer.py
+++ b/text_recognizer/models/transformer.py
@@ -1,12 +1,9 @@
"""PyTorch Lightning model for base Transformers."""
from typing import Dict, List, Optional, Union, Tuple, Type
-from omegaconf import DictConfig, OmegaConf
-import pytorch_lightning as pl
-import torch
+from omegaconf import DictConfig
from torch import nn
from torch import Tensor
-import torch.nn.functional as F
import wandb
from text_recognizer.data.emnist import emnist_mapping
diff --git a/text_recognizer/models/vqvae.py b/text_recognizer/models/vqvae.py
index 9857420..18e8691 100644
--- a/text_recognizer/models/vqvae.py
+++ b/text_recognizer/models/vqvae.py
@@ -1,10 +1,9 @@
"""PyTorch Lightning model for base Transformers."""
from typing import Any, Dict, Union, Tuple, Type
-from omegaconf import DictConfig, OmegaConf
+from omegaconf import DictConfig
from torch import nn
from torch import Tensor
-import torch.nn.functional as F
import wandb
from text_recognizer.models.base import LitBaseModel
@@ -35,7 +34,12 @@ class LitVQVAEModel(LitBaseModel):
"""Logs prediction on image with wandb."""
try:
self.logger.experiment.log(
- {title: [wandb.Image(data[0]), wandb.Image(reconstructions[0]),]}
+ {
+ title: [
+ wandb.Image(data[0]),
+ wandb.Image(reconstructions[0]),
+ ]
+ }
)
except AttributeError:
pass