Basic Framework

This commit is contained in:
hofee
2024-08-18 00:37:17 +08:00
commit 73dcd592df
14 changed files with 733 additions and 0 deletions

View File

@@ -0,0 +1,7 @@
EXTERNAL_FREEZE_MODULES = set()
def external_freeze(cls):
if not hasattr(cls, 'load') or not callable(getattr(cls, 'load')):
raise TypeError(f"external module <{cls.__name__}> must implement a 'load' method")
EXTERNAL_FREEZE_MODULES.add(cls)
return cls

34
annotations/stereotype.py Normal file
View File

@@ -0,0 +1,34 @@
# --- Classes --- #
def dataset():
pass
def module():
pass
def pipeline():
pass
def runner():
pass
def factory():
pass
# --- Functions --- #
evaluation_methods = {}
def evaluation_method(eval_type):
def decorator(func):
evaluation_methods[eval_type] = func
return func
return decorator
def loss_function():
pass
# --- Main --- #