提交 e687af48 编写于 作者: Z Zeyu Chen

add senta test

上级 7b5c4a66
......@@ -91,6 +91,8 @@ def parse_args():
def remove_feed_fetch_op(program):
""" remove feed and fetch operator and variable for fine-tuning
"""
print("remove feed fetch op")
block = program.global_block()
need_to_remove_op_index = []
......@@ -168,8 +170,10 @@ def train_net(train_reader,
main_program = fluid.Program()
startup_program = fluid.Program()
# use switch program to test fine-tuning
fluid.framework.switch_main_program(module.get_inference_program())
# remove feed fetch operator and variable
remove_feed_fetch_op(fluid.default_main_program())
label = fluid.layers.data(name="label", shape=[1], dtype="int64")
......
[metadata]
license_file = LICENSE
# Copyright 2018 The TensorFlow Hub 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.
# ==============================================================================
"""Setup for pip package."""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from setuptools import find_packages
from setuptools import setup
__version__ = "0.1.0-dev"
REQUIRED_PACKAGES = [
'numpy >= 1.12.0',
'six >= 1.10.0',
'protobuf >= 3.4.0',
]
setup(
name='paddle_hub',
version=__version__.replace('-', ''),
description=('PaddleHub is a library to foster the publication, '
'discovery, and consumption of reusable parts of machine '
'learning models.'),
long_description='',
url='https://github.com/PaddlePaddle/PaddleHub',
author='PaddlePaddle Author',
author_email='chenzeyu01@baidu.com',
install_requires=REQUIRED_PACKAGES,
packages=find_packages(),
# PyPI package information.
classifiers=[
'Development Status :: 4 - Beta',
'Intended Audience :: Developers',
'Intended Audience :: Education',
'Intended Audience :: Science/Research',
'License :: OSI Approved :: Apache Software License',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.4',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Topic :: Scientific/Engineering',
'Topic :: Scientific/Engineering :: Mathematics',
'Topic :: Scientific/Engineering :: Artificial Intelligence',
'Topic :: Software Development',
'Topic :: Software Development :: Libraries',
'Topic :: Software Development :: Libraries :: Python Modules',
],
license='Apache 2.0',
keywords=('paddlepaddle pretrained paddle-hub'),
)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册