Esempio n. 1
0
# flake8: noqa
from catalyst import registry
from catalyst.dl import SupervisedRunner as Runner

from .experiment import Experiment
from .net import Net

registry.Model(Net)
Esempio n. 2
0
# flake8: noqa
from catalyst import registry
from catalyst.dl import SupervisedRunner as Runner

from .experiment import Experiment
from .model import SimpleNet

registry.Model(SimpleNet)
Esempio n. 3
0
# flake8: noqa
# pylint: disable=unused-import

from catalyst.contrib.models.nlp import BertClassifier
from catalyst import registry
from catalyst.dl import SupervisedRunner as Runner

from .experiment import Experiment

registry.Model(BertClassifier)