Commit bfd2c328 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add __model_name__ and __dims__ to model

parent e4962df5

zoo/monochrome/metaformer.py

+2 −0
Original line number Diff line number Diff line
@@ -29,6 +29,8 @@ class CAFormerBuilder: 


    def __call__(self, *args, **kwargs):

        model = create_model(**self.create_model_args)

        model.__model_name__ = self.__model_name__

        model.__dims__ = self.__dims__

        return model