diff --git a/facer/farl/model.py b/facer/farl/model.py index 7fd5786..4ad1d0d 100644 --- a/facer/farl/model.py +++ b/facer/farl/model.py @@ -8,7 +8,7 @@ from torch import nn import torch.utils.checkpoint as checkpoint import numpy as np -from timm.models.layers import trunc_normal_, DropPath +from timm.layers import trunc_normal_, DropPath class Bottleneck(nn.Module):