Commit 0e164f43 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): process new ckpts

parent 7a2b9d43

zoo/monochrome/__main__.py

+3 −1
Original line number Diff line number Diff line
@@ -82,7 +82,9 @@ _KNOWN_CKPTS: List[Tuple[str, str, int]] = [ 
    # ('monochrome-resnet18-safe2-450.ckpt', 'resnet18', 180),

    # ('monochrome-levit_d0.2-500.ckpt', 'levit', 180),

    # ('monochrome-caformer_safe2-80.ckpt', 'caformer', 180),

    ('monochrome-caformer_safe4-70.ckpt', 'caformer', 180),

    # ('monochrome-caformer_safe4-70.ckpt', 'caformer', 180),

    ('monochrome-caformer-40.ckpt', 'caformer', 180),

    ('monochrome-caformer-110.ckpt', 'caformer', 180),

]