Duplicated fromΒ yisol/IDM-VTON
938e515
1
2
3
4
5
6
from .bn import ABN, InPlaceABN, InPlaceABNSync from .functions import ACT_RELU, ACT_LEAKY_RELU, ACT_ELU, ACT_NONE from .misc import GlobalAvgPool2d, SingleGPU from .residual import IdentityResidualBlock from .dense import DenseModule