From 34d11be58c971379f4488922fd9d853174a2bd79 Mon Sep 17 00:00:00 2001 From: Jeff Rhoades <37990507+rhoadesScholar@users.noreply.github.com> Date: Wed, 20 Mar 2024 09:42:23 -0400 Subject: [PATCH] Update start.py --- dacapo/experiments/starts/start.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dacapo/experiments/starts/start.py b/dacapo/experiments/starts/start.py index 6204c56fd..6c1622031 100644 --- a/dacapo/experiments/starts/start.py +++ b/dacapo/experiments/starts/start.py @@ -13,8 +13,8 @@ def match_heads(model, head_weights, old_head, new_head ): new_index = new_head.index(label) for key in head_keys: if key in model.state_dict().keys(): - n_val = head_weights[key][old_index] - model.state_dict()[key][new_index] = n_val + new_value = head_weights[key][old_index] + model.state_dict()[key][new_index] = new_value logger.warning(f"matched head for {label}.") def _set_weights(model, weights, run, criterion, old_head=None, new_head=None):