multiml.agent.pytorch.pytorch_connection_random_search module

class multiml.agent.pytorch.pytorch_connection_random_search.PytorchConnectionRandomSearchAgent(freeze_model_weights=False, do_pretraining=True, connectiontask_name=None, connectiontask_args={}, **kwargs)

Bases: ConnectionRandomSearchAgent

Pytorch implementation for ConnectionRandomSearchAgent.

class ModelConnectionTask(subtasks, loss_weights=None, variable_mapping=None, **kwargs)

Bases: ModelConnectionTask, PytorchBaseTask

Pytorch implementation of ModelConnectionTask.

build_model(): Build model.