30 lines
834 B
Python
30 lines
834 B
Python
import paddle
|
|
from paddle.nn import Sigmoid
|
|
from paddle.nn import Tanh
|
|
from ..legendary_models.pp_lcnet import PPLCNet_x2_5
|
|
|
|
__all__ = ["PPLCNet_x2_5_Tanh"]
|
|
|
|
|
|
class TanhSuffix(paddle.nn.Layer):
|
|
def __init__(self, origin_layer):
|
|
super(TanhSuffix, self).__init__()
|
|
self.origin_layer = origin_layer
|
|
self.tanh = Tanh()
|
|
|
|
def forward(self, input, res_dict=None, **kwargs):
|
|
x = self.origin_layer(input)
|
|
x = self.tanh(x)
|
|
return x
|
|
|
|
|
|
def PPLCNet_x2_5_Tanh(pretrained=False, use_ssld=False, **kwargs):
|
|
def replace_function(origin_layer, pattern):
|
|
new_layer = TanhSuffix(origin_layer)
|
|
return new_layer
|
|
|
|
pattern = "fc"
|
|
model = PPLCNet_x2_5(pretrained=pretrained, use_ssld=use_ssld, **kwargs)
|
|
model.upgrade_sublayer(pattern, replace_function)
|
|
return model
|