readablePTH 0.1.3

Make pth readable

Readable PTH

About this project

Make pth readable.
model.pth <--> model.json

Installation

pip install readablePTH

For develop

Please clone the project

git clone [e-mail]:oneview-space/readablePTH.git

And then install

cd readablePTH
pip install -e .

Usage

For now this is only working with [encode, decode] commands

decode from pth to json

rpth decode -i model.pth -o model.json 

encode from json to pth

rpth encode -i model.json -o model.pth 

TODO:

• Add the ability to use any torch version
• add args to map_location (CPU/GPU) when writing pth file