dacapo.store.local_weights_store

Module Contents

Classes

LocalWeightsStore

A local store for network weights.

Attributes

logger

dacapo.store.local_weights_store.logger

class dacapo.store.local_weights_store.LocalWeightsStore(basedir)

A local store for network weights.

latest_iteration(run: str) → int | None

Return the latest iteration for which weights are available for the given run.

store_weights(run: dacapo.experiments.run.Run, iteration: int)

Store the network weights of the given run.

retrieve_weights(run: str, iteration: int) → dacapo.store.weights_store.Weights

Retrieve the network weights of the given run.

remove(run: str, iteration: int)

Delete the weights associated with a specific run/iteration

store_best(run: str, iteration: int, dataset: str, criterion: str)

Store the best weights in a easy to find location. Symlinks weights from appropriate iteration # TODO: simply store a yaml of dataset/criterion -> iteration/parameter id

retrieve_best(run: str, dataset: str | dacapo.experiments.datasplits.datasets.dataset.Dataset, criterion: str) → int

Retrieve the best weights for this run/dataset/criterion