Update davit.py

3 years ago · cf1cc8b1d1
parent 0ed0e9ac35
commit cf1cc8b1d1
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -23,7 +23,7 @@ from typing import Any, Dict, Iterable, Iterator, Mapping, Optional, overload, T
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-import torch.Tensor as Tensor
+from torch import Tensor
 from .helpers import build_model_with_cfg
 from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 from .layers import DropPath, to_2tuple, trunc_normal_, SelectAdaptivePool2d, ClassifierHead, Mlp