Commit ff54af8b authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): update export new model

parent 4c393602

zoo/ccip/__main__.py

+4 −3
Original line number Diff line number Diff line
@@ -173,9 +173,10 @@ def onnx_check(model: str, verbose: bool = False, 




MODELS = [

    ('caformer', 'ccip-caformer-2_fp32.ckpt'),

    ('caformer', 'ccip-caformer-4_fp32.ckpt'),

    ('caformer', 'ccip-caformer-5_fp32.ckpt'),

    # ('caformer', 'ccip-caformer-2_fp32.ckpt'),

    # ('caformer', 'ccip-caformer-4_fp32.ckpt'),

    # ('caformer', 'ccip-caformer-5_fp32.ckpt'),

    ('caformer', 'ccip-caformer-23_randaug_fp32.ckpt'),

]