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):
