dacapo.experiments.starts.cosem_start

Module Contents

Classes

CosemStart

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

Functions

get_model_setup(run)

Attributes

logger

dacapo.experiments.starts.cosem_start.logger

dacapo.experiments.starts.cosem_start.get_model_setup(run)

class dacapo.experiments.starts.cosem_start.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.