dacapo.experiments.starts.start

Module Contents

Classes

Start

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

Functions

match_heads(model, head_weights, old_head, new_head)

Attributes

logger
head_keys

dacapo.experiments.starts.start.logger

dacapo.experiments.starts.start.head_keys = ['prediction_head.weight', 'prediction_head.bias', 'chain.1.weight', 'chain.1.bias']

dacapo.experiments.starts.start.match_heads(model, head_weights, old_head, new_head)

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