multiml.task.pytorch.pytorch_model_connection module

Pytorch ModelConnectionTask module.

class multiml.task.pytorch.pytorch_model_connection.ModelConnectionTask(subtasks, loss_weights=None, variable_mapping=None, **kwargs)

Bases: ModelConnectionTask, PytorchBaseTask

Pytorch implementation of ModelConnectionTask.

build_model()

Build model.