Commit 4f932739 authored by dzy7e's avatar dzy7e
Browse files

caformer

parent 04cacf27

zoo/monochrome/train_.py

+1 −1
Original line number Diff line number Diff line
@@ -55,7 +55,7 @@ _register_model(ResNet1522D) 
_register_model(SigTransformer)

_register_model(LeSigTransformer)

_register_model(LeViT)

_register_model(CAFormerBuilder)

_register_model(CAFormerBuilder())





def _register_dataset(cls: Type[Dataset]):