From 6391c90b81ff2ef9785c14278017ce0aeaff8a88 Mon Sep 17 00:00:00 2001 From: Bin Lu Date: Tue, 1 Jun 2021 11:35:44 +0800 Subject: [PATCH] Delete ppcls/arch/neck directory --- ppcls/arch/neck/__init__.py | 23 ----------------------- ppcls/arch/neck/fc.py | 33 --------------------------------- 2 files changed, 56 deletions(-) delete mode 100644 ppcls/arch/neck/__init__.py delete mode 100644 ppcls/arch/neck/fc.py diff --git a/ppcls/arch/neck/__init__.py b/ppcls/arch/neck/__init__.py deleted file mode 100644 index 93529b72..00000000 --- a/ppcls/arch/neck/__init__.py +++ /dev/null @@ -1,23 +0,0 @@ -# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -__all__ = ['build_neck"] - -def build_neck(config): - support_dict = ['FPN', 'FC'] - module_name = config.pop('name') - assert module_name in support_dict, Exception('neck only support {}'.format( - support_dict)) - module_class = eval(module_name)(**config) - return module_class diff --git a/ppcls/arch/neck/fc.py b/ppcls/arch/neck/fc.py deleted file mode 100644 index e22961b6..00000000 --- a/ppcls/arch/neck/fc.py +++ /dev/null @@ -1,33 +0,0 @@ -# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - -from __future__ import absolute_import -from __future__ import division -from __future__ import print_function - -import paddle -import paddle.nn as nn - -class FC(nn.Layer): - def __init__(self, input_dim, - embedding_size): - super(FC, self).__init__() - self.input_dim = input_dim - self.embedding_size = embedding_size - weight_attr = paddle.ParamAttr(initializer = paddle.nn.initializer.XavierNormal()) - self.fc = paddle.nn.Linear(self.input_dim, self.embedding_size, weight_attr=weight_attr) - - def forward(self, x): - x = self.fc(x) - return x -- GitLab