from core.leras import nn class ArchiBase(): def __init__(self, *args, name=None, **kwargs): self.name=name #overridable def flow(self, *args, **kwargs): raise Exception("this archi does not support flow. Use model classes directly.") #overridable def get_weights(self): pass nn.ArchiBase = ArchiBase