Commit 542e2938 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): dump 450 to onnx

parent 0fdbfd4e

zoo/monochrome/__main__.py

+1 −1
Original line number Diff line number Diff line
@@ -43,7 +43,7 @@ _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),

    ('monochrome-resnet18-safe2-290.ckpt', 'resnet18', 180),

    ('monochrome-resnet18-safe2-450.ckpt', 'resnet18', 180),

]