1
"""PyTorch-based multi-task learning framework for discriminative modeling."""
2

3 2
from .data import DictDataLoader, DictDataset  # noqa: F401
4 2
from .loss import cross_entropy_with_probs  # noqa: F401
5 2
from .multitask_classifier import MultitaskClassifier  # noqa: F401
6 2
from .task import Operation, Task  # noqa: F401
7 2
from .training.loggers import (  # noqa: F401
8
    Checkpointer,
9
    CheckpointerConfig,
10
    LogManager,
11
    LogManagerConfig,
12
    LogWriter,
13
    LogWriterConfig,
14
    TensorBoardWriter,
15
)
16 2
from .training.trainer import Trainer  # noqa: F401

Read our documentation on viewing source code .

Loading