Commit e4962df5 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): export caformer model

parent 78339643

.github/workflows/export.yml

+1 −1
Original line number Diff line number Diff line
@@ -16,7 +16,7 @@ jobs: 
        python-version:

          - '3.8'

        model-name:

          - 'lpips'

          #          - 'lpips'

          - 'monochrome'



    steps:

zoo/monochrome/__main__.py

+2 −1
Original line number Diff line number Diff line
@@ -80,7 +80,8 @@ _KNOWN_CKPTS: List[Tuple[str, str, int]] = [ 
    # ('monochrome-resnet18-480.ckpt', 'resnet18', 180),

    # ('monochrome-transformer-480.ckpt', 'transformer', 180),

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

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

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

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

]