Commit 2f4f0770 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): try dump models

parent cd5f9ec9

.github/workflows/export.yml

+1 −0
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@ jobs: 
          - '3.8'

        model-name:

          - 'lpips'

          - 'monochrome'



    steps:

      - name: Checkout code

zoo/monochrome/__main__.py

+1 −1
Original line number Diff line number Diff line
@@ -40,7 +40,7 @@ def export_one(output: str, feature_bins: int, ckpt: str, model_name: str): 




_KNOWN_CKPTS: List[Tuple[str, str, int]] = [

    ('monochrome-alexnet_x-400.ckpt', 'alexnet', 256),

    ('monochrome-alexnet_plus-320.ckpt', 'alexnet', 256),

]