Commit 9e7406aa authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): make transformer's onnx

parent 773c57ff
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -42,6 +42,7 @@ def export_one(output: str, feature_bins: int, ckpt: str, model_name: str):
_KNOWN_CKPTS: List[Tuple[str, str, int]] = [
    ('monochrome-alexnet-480.ckpt', 'alexnet', 180),
    ('monochrome-resnet18-480.ckpt', 'resnet18', 180),
    ('monochrome-transformer-480.ckpt', 'transformer', 180),
]