From c48e15796b1355cf6409c96ea87882811960dbff Mon Sep 17 00:00:00 2001 From: ifrit98 Date: Sun, 1 Oct 2023 18:41:24 +0000 Subject: [PATCH] handle case where mg shrinks from previously stored chain state --- prompting/validators/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/prompting/validators/utils.py b/prompting/validators/utils.py index 5b1101c..f39daed 100644 --- a/prompting/validators/utils.py +++ b/prompting/validators/utils.py @@ -128,7 +128,8 @@ def resync_metagraph(self: "validators.neuron.neuron"): if len(self.hotkeys) < len(self.metagraph.hotkeys): # Update the size of the moving average scores. new_moving_average = torch.zeros((self.metagraph.n)).to(self.device) - new_moving_average[: len(self.hotkeys)] = self.moving_averaged_scores + min_len = min(len(self.hotkeys), len(self.moving_averaged_scores)) + new_moving_average[:min_len] = self.moving_averaged_scores[:min_len] self.moving_averaged_scores = new_moving_average # Resize the gating model.