未验证 提交 2b84db06 编写于 作者: B Bin Lu 提交者: GitHub

Update mobilenet_v1.py

上级 084a27a1
...@@ -21,7 +21,6 @@ import paddle.nn as nn ...@@ -21,7 +21,6 @@ import paddle.nn as nn
from paddle.nn import Conv2D, BatchNorm, Linear, Dropout, ReLU, Flatten from paddle.nn import Conv2D, BatchNorm, Linear, Dropout, ReLU, Flatten
from paddle.nn import AdaptiveAvgPool2D from paddle.nn import AdaptiveAvgPool2D
from paddle.nn.initializer import KaimingNormal from paddle.nn.initializer import KaimingNormal
import math
from ppcls.arch.backbone.base.theseus_layer import TheseusLayer from ppcls.arch.backbone.base.theseus_layer import TheseusLayer
from ppcls.utils.save_load import load_dygraph_pretrain_from, load_dygraph_pretrain_from_url from ppcls.utils.save_load import load_dygraph_pretrain_from, load_dygraph_pretrain_from_url
...@@ -106,7 +105,6 @@ class MobileNet(TheseusLayer): ...@@ -106,7 +105,6 @@ class MobileNet(TheseusLayer):
super(MobileNet, self).__init__() super(MobileNet, self).__init__()
self.scale = scale self.scale = scale
self.pretrained = pretrained self.pretrained = pretrained
self.block_list = []
self.conv = ConvBNLayer( self.conv = ConvBNLayer(
num_channels=3, num_channels=3,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册