Remove unused imports

dataset/base/__init__.py

@@ -17,7 +17,6 @@
 import os.path as osp
 from typing import Any, Dict, List, Tuple, Union
 
-import torch
 from hydra.utils import get_original_cwd
 from torch import Tensor
 from torch.utils.data import Dataset

dataset/base/image.py

@@ -12,7 +12,6 @@
 import abc
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
-import torch
 from torch import Tensor
 from torchvision.io.image import read_image  # type: ignore
 from torchvision.transforms import transforms  # type: ignore

src/base_tester.py

@@ -13,7 +13,6 @@
 from collections import defaultdict
 from typing import Any, Dict, List, Optional, Tuple, TypeVar, Union
 
-import torch
 from torch import Tensor
 from torch.nn import Module
 from torch.utils.data import DataLoader

utils/training.py

@@ -12,7 +12,6 @@
 
 from typing import List, Tuple, Union
 
-import torch
 from torch import Tensor, nn
 
 import conf.project as project_conf