Commit 04cacf27 authored by dzy7e's avatar dzy7e
Browse files

caformer

parent 3f5c288e

zoo/monochrome/metaformer.py

+2 −0
Original line number Diff line number Diff line
@@ -5,6 +5,8 @@ import zoo.monochrome.metaformer_timm # register models 


class CAFormerBuilder:

    __model_name__ = 'caformer'

    __dims__ = 2



    def __init__(self, arch='caformer_m36_384_in21ft1k', num_classes=2, drop_path_rate=0.4):

        self.create_model_args = dict(

            model_name=arch,