Skip to content

Commit

Permalink
Merge pull request #37 from janelia-cellmap/actions/black
Browse files Browse the repository at this point in the history
Format Python code with psf/black push
  • Loading branch information
mzouink authored Feb 9, 2024
2 parents 910b2e0 + 906dfd6 commit 8cce986
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions dacapo/experiments/starts/start.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}")
Expand All @@ -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")


0 comments on commit 8cce986

Please sign in to comment.