资源简介

pytorch的机制便于快速开发模型,但是在产品上的运用不稳定,需要将其转换为keras对应的模型权重,使用该代码进行转换的示例可见:https://blog.csdn.net/xiaoxifei/article/details/82685298

资源截图

代码片段和文件信息

from setuptools import setup

setup(
    name=‘nn_transfer‘
    version=‘0.1.0‘
    description=‘Transfer weights between Keras and PyTorch.‘
    install_requires=[
        ‘numpy‘
        ‘keras‘
        ‘h5py‘
    ]
    test_suite=‘nose.collector‘
    tests_require=[‘nose‘]
    packages=[‘nn_transfer‘]


 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-12-28 02:26  nn-transfer-master\
     文件        1187  2017-12-28 02:26  nn-transfer-master\.gitignore
     文件        1408  2017-12-28 02:26  nn-transfer-master\.travis.yml
     文件        1070  2017-12-28 02:26  nn-transfer-master\LICENSE
     文件        1683  2017-12-28 02:26  nn-transfer-master\README.md
     文件       11258  2017-12-28 02:26  nn-transfer-master\example.ipynb
     目录           0  2017-12-28 02:26  nn-transfer-master\nn_transfer\
     文件           0  2017-12-28 02:26  nn-transfer-master\nn_transfer\__init__.py
     目录           0  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\
     文件           0  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\__init__.py
     目录           0  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\architectures\
     文件           0  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\architectures\__init__.py
     文件        1595  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\architectures\lenet.py
     文件        1237  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\architectures\simplenet.py
     文件        6957  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\architectures\unet.py
     文件        2435  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\architectures\vggnet.py
     文件        1780  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\helpers.py
     文件        1890  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\test_architectures.py
     文件        6139  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\test_layers.py
     文件         476  2017-12-28 02:26  nn-transfer-master\nn_transfer\test\test_util.py
     文件        7643  2017-12-28 02:26  nn-transfer-master\nn_transfer\transfer.py
     文件         864  2017-12-28 02:26  nn-transfer-master\nn_transfer\util.py
     文件         318  2017-12-28 02:26  nn-transfer-master\setup.py

评论

共有 条评论