dacapo.experiments.tasks.pretrained_task

Classes

PretrainedTask

A task that uses a pretrained model. The model is loaded from a file

Module Contents

class dacapo.experiments.tasks.pretrained_task.PretrainedTask(task_config)

A task that uses a pretrained model. The model is loaded from a file and the weights are loaded from a file.

weights

The path to the weights file.

Type:

Path

create_model(self, architecture) → Model

This method creates a model using the given architecture.

Notes

This is a base class for all tasks that use pretrained models.

sub_task

weights

predictor

loss

post_processor

evaluator

create_model(architecture)

Create a model using the given architecture.

Parameters:

architecture (str) – The architecture of the model.

Returns:

The model created using the given architecture.

Return type:

Model

Raises:

NotImplementedError – This method is not implemented.

Examples

>>> model = task.create_model(architecture)