fix shop_segmentation to use old timm lib

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/14443535
* fix shop_segmentation to use old timm lib
This commit is contained in:
wenmeng.zwm
2023-10-26 13:24:06 +08:00
parent e8b34a3bcd
commit ae00f39b44
2 changed files with 4 additions and 4 deletions

View File

@@ -11,8 +11,8 @@ from collections import OrderedDict
import torch
import torch.nn.functional as F
import torch.utils.checkpoint as checkpoint
from timm.layers.drop import drop_path
from timm.layers.weight_init import trunc_normal_
from timm.models.layers.drop import drop_path
from timm.models.layers.weight_init import trunc_normal_
from torch import nn

View File

@@ -8,8 +8,8 @@
import torch.nn as nn
import torch.nn.functional as F
from mmcv.cnn import ConvModule
from timm.layers.drop import drop_path
from timm.layers.weight_init import trunc_normal_
from timm.models.layers.drop import drop_path
from timm.models.layers.weight_init import trunc_normal_
from .common import resize