8000 GitHub - Sehun0819/pytea: PyTEA: PyTorch Tensor Error Analyzer
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

Sehun0819/pytea

 
 

Repository files navigation

Torch2H: PyTorch Graph -> H Program

Requirements

  • node.js >= 12.x
  • python >= 3.8

How to build and use

# install dependencies
npm run install:all

# build
npm run build

# running example: Generates H program "out.h".
python torch2h.py examples/mnist/main.py

TODO

  • Make realistic weights by learning network. Current weights are just randomly generated floats.
  • Add more benchmarks and functions accordingly.

Etc.

This program has built on PyTea.

About

PyTEA: PyTorch Tensor Error Analyzer

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • TypeScript 58.7%
  • Python 41.0%
  • Other 0.3%
0