from omnibelt import InitWall, unspecified_argument, Simple_Child, get_printer, Named import omnifig as fig from wrapt import ObjectProxy import numpy as np prt = get_printer(__file__) try: import timm except ImportError: prt.warning('Unable to import "timm"') from .. import util from .framework import Function, FunctionWrapper, Encodable @fig.AutoModifier('extractor') class Extractor(Named, Encodable, Function): pass @fig.Component('_pretrained') def build_pretrained(A, ID=None, pretrained=None, model_kwargs=None, include_config_args=None): if ID is None: ID = A.pull('model-ID', '<>ident')
'allow_profile_sources': True, 'use_last_entry_as_default': True, 'default_ptype': 'default', # 'resolution_order': [ # for scripts, configs, components, and modifiers # 'local', # check the current working dir if its in a project # 'active-project', # currently active project # 'related-projects', # any projects related to the active project # 'init' # scripts loaded through init (no project) # ], 'arg_parse_language': 'yaml', }) prt = get_printer(__name__) def configurize(data): ''' Transform data container to use config objects (ConfigDict/ConfigList) :param data: dict/list data :return: deep copy of data using ConfigDict/ConfigList ''' if isinstance(data, global_settings['config_type']): return data for typ, convert in global_settings.get('config_converters', {}).items(): allow_subtypes = True if isinstance(convert, tuple):