diff --git a/dacapo/experiments/starts/start.py b/dacapo/experiments/starts/start.py index d1561ed0..70f77e31 100644 --- a/dacapo/experiments/starts/start.py +++ b/dacapo/experiments/starts/start.py @@ -11,6 +11,7 @@ def __init__(self, start_config): def initialize_weights(self, model): from dacapo.store.create_store import create_weights_store + weights_store = create_weights_store() weights = weights_store._retrieve_weights(self.run, self.criterion) logger.info(f"loading weights from run {self.run}, criterion: {self.criterion}") @@ -27,5 +28,3 @@ def initialize_weights(self, model): model_dict[layer] = weights.model[layer] model.load_state_dict(model_dict) logger.warning(f"loaded only common layers from weights") - -