Update metaformers.py

pull/1647/head
Fredo Guan 2 years ago
parent 1f3a661160
commit d17bcb10a5

@ -21,7 +21,7 @@ from functools import partial
import torch
import torch.nn as nn
from timm.layers import trunc_normal_, DropPath
from timm.models.registry import register_model
from ._registry import register_model
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
from timm.layers.helpers import to_2tuple

Loading…
Cancel
Save