48cafca
1
2
3
4
5
6
7
8
from .vit import vit, vitl def create_backbone(cfg): if cfg.MODEL.BACKBONE.TYPE == 'vit': return vit(cfg) else: raise NotImplementedError('Backbone type is not implemented')
