Skip to content

Instantly share code, notes, and snippets.

View victor-shepardson's full-sized avatar

Victor Shepardson victor-shepardson

View GitHub Profile
@victor-shepardson
victor-shepardson / unwrap.neutone.py
Created February 14, 2024 13:55
unwrap nn~ models from neutone models
from pathlib import Path
from fire import Fire
import torch
def main(model:str, outfile:str=None):
"""
Args:
model: path to neutone model file
outfile: path to write nn~ model file