BerkeleyLab / nexport

A Python package for exporting the weights and biases of neural networks.

Home Page:https://pypi.org/project/nexport

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool


GitHub tag (latest by date) GitHub release (latest by date) GitHub Release Date GitHub license GitHub commit activity GitHub wiki

PyPI PyPI - Python Version PyPI - Wheel PyPI - Status PyPI - Downloads

GitHub Repo stars GitHub watchers GitHub forks Lines of code GitHub repo file count GitHub repo size

Overview

nexport is a lightweight Python 3.10+ package which empowers deep learning developers in exporting the trainable parameters of deep neural networks to human-readable and transrerable file types.

Table of contents

Current support

Filetype PyTorch Keras/TensorFlow
Export Import Export Import
Text (.txt) βœ… βœ… 🚧 ❌
JSON (.json) βœ… 🚧 ❌ ❌
CSV (.csv) ❌ ❌ ❌ ❌
XML (.xml) ❌ ❌ ❌ ❌

Install & use

  1. From terminal: pip install nexport
  2. From python environment: import nexport

Objectives

  • Export weights and biases to human-readable file
  • Ensure compatability with all popular neural network development software

History

This package is intended to be used in conjunction with inference-engine. As such, nexport was developed by the inference-engine developers to enable compatability between the two softwares. nexport does this by exporting the weights and biases from networks compiled in PyTorch, Keras, and TensorFlow into standardized human-readable files. These files can be read by inference-engine to instantiate the netwoks in Fortran 2018 for inference.

Credits

nexport was created and is currently maintained by Jordan Welsman. Parts of this project were based on prior work by Tan Nguyen.

License

nexport is developed and distributed under a Berkeley Laboratory modified BSD license.

Note See LICENSE for more details.

Links

πŸ“ See this project on GitHub

🎁 See this project on PyPI

🐱 Follow me on GitHub

πŸ’Ό Connect with me on Linkedin

πŸ“§ Send me an email

πŸ’­ Based on this project

About

A Python package for exporting the weights and biases of neural networks.

https://pypi.org/project/nexport

License:Other


Languages

Language:Python 98.5%Language:Shell 1.5%