dacapo.experiments.starts

Submodules

Package Contents

Classes

Start

This class interfaces with the dacapo store to retrieve and load the

StartConfig

A class to represent the configuration for running tasks.

CosemStart

This class interfaces with the dacapo store to retrieve and load the

CosemStartConfig

Starter for COSEM pretained models. This is a subclass of StartConfig and
class dacapo.experiments.starts.Start(start_config)

This class interfaces with the dacapo store to retrieve and load the weights of the starter model used for finetuning.

run

The specified run to retrieve weights for the model.

Type:

str

criterion

The policy that was used to decide when to store the weights.

Type:

str

initialize_weights(model, new_head=None)

Retrieves the weights from the dacapo store and load them into the model. :param model: The model to which the weights are to be loaded. :type model: obj

Raises:

RuntimeError – If weights of a non-existing or mismatched layer are being loaded, a RuntimeError exception is thrown which is logged and handled by loading only the common layers from weights.

class dacapo.experiments.starts.StartConfig

A class to represent the configuration for running tasks.

run

The run to be used as a starting point for tasks.

Type:

str

criterion

The criterion to be used for choosing weights from run.

Type:

str

start_type
run: str
criterion: str
class dacapo.experiments.starts.CosemStart(start_config)

This class interfaces with the dacapo store to retrieve and load the weights of the starter model used for finetuning.

run

The specified run to retrieve weights for the model.

Type:

str

criterion

The policy that was used to decide when to store the weights.

Type:

str

check()
initialize_weights(model, new_head=None)

Retrieves the weights from the dacapo store and load them into the model. :param model: The model to which the weights are to be loaded. :type model: obj

Raises:

RuntimeError – If weights of a non-existing or mismatched layer are being loaded, a RuntimeError exception is thrown which is logged and handled by loading only the common layers from weights.

class dacapo.experiments.starts.CosemStartConfig

Starter for COSEM pretained models. This is a subclass of StartConfig and should be used to initialize the model with pretrained weights from a previous run.

start_type