Skip to content

How to import models from several `nnfabrik` pipelines.

Fabian Sinz edited this page Feb 23, 2021 · 1 revision
import datajoint as dj
dj.config["enable_python_native_blobs"] = True

from nnfabrik.templates import TrainedModelBase
from nnfabrik.main import my_nnfabrik

my_nnfabrik_1 = my_nnfabrik('nnfabrik_konsti_v1', use_common_fabrikant=False)
my_nnfabrik_2 = my_nnfabrik('nnfabrik_konsti_v4', use_common_fabrikant=False)

@my_nnfabrik_1.schema
class TrainedModel(TrainedModelBase):
    nnfabrik = my_nnfabrik_1

@my_nnfabrik_2.schema
class TrainedModel(TrainedModelBase):
    nnfabrik = my_nnfabrik_2
Clone this wiki locally