How to use potato2chip:
- Save your PyTorch model using TorchScript to your working directory (recommended method):
model_scripted = torch.jit.script(model) # Export to TorchScript model_scripted.save('model_scripted.pt') # Save
- Create a script that loads your saved model, and inside it, import
converter.py
:import converter ... model = torch.jit.load('model_scripted.pt')
- Now add the line
converter.gen_pyrtl(model)
- Execute your script
- You will see the outputed pyrtl code in 'output.py'