Update metaformers.py

pull/1647/head
Fredo Guan 2 years ago
parent 7f149f31d4
commit 7aa3459caf

@ -17,6 +17,7 @@ MetaFormer baselines including IdentityFormer, RandFormer, PoolFormerV2,
ConvFormer and CAFormer.
Some implementations are modified from timm (https://github.com/rwightman/pytorch-image-models).
"""
from collections import OrderedDict
from functools import partial
import torch
import torch.nn as nn

Loading…
Cancel
Save