Update qlnet.py
Browse files
qlnet.py
CHANGED
@@ -15,12 +15,12 @@ import torch
|
|
15 |
import torch.nn as nn
|
16 |
import torch.nn.functional as F
|
17 |
|
18 |
-
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
|
19 |
-
from timm.layers import DropBlock2d, DropPath, AvgPool2dSame, BlurPool2d, GroupNorm, create_attn, get_attn, \
|
20 |
-
|
21 |
|
22 |
-
from ._builder import build_model_with_cfg
|
23 |
-
from ._registry import register_model, model_entrypoint
|
24 |
|
25 |
def get_padding(kernel_size, stride, dilation=1):
|
26 |
padding = ((stride - 1) + dilation * (kernel_size - 1)) // 2
|
|
|
15 |
import torch.nn as nn
|
16 |
import torch.nn.functional as F
|
17 |
|
18 |
+
# from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
|
19 |
+
# from timm.layers import DropBlock2d, DropPath, AvgPool2dSame, BlurPool2d, GroupNorm, create_attn, get_attn, \
|
20 |
+
# get_act_layer, get_norm_layer, create_classifier, LayerNorm2d
|
21 |
|
22 |
+
# from ._builder import build_model_with_cfg
|
23 |
+
# from ._registry import register_model, model_entrypoint
|
24 |
|
25 |
def get_padding(kernel_size, stride, dilation=1):
|
26 |
padding = ((stride - 1) + dilation * (kernel_size - 1)) // 2
|