diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-27 20:27:25 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-27 20:27:25 +0200 |
| commit | 6f21058f55ab9146673cf0098dc27869aee4090c (patch) | |
| tree | 5287104e126b0fe65558b094f6a58569655bc03a /text_recognizer/models/transformer.py | |
| parent | d08a9f87a86ae6ed4a1077a3038fc444085d7f6e (diff) | |
Remove unused imports
Diffstat (limited to 'text_recognizer/models/transformer.py')
| -rw-r--r-- | text_recognizer/models/transformer.py | 5 |
1 files changed, 1 insertions, 4 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 |