From bc8eaf0825b33bc8bd8318d8a6d512d308216329 Mon Sep 17 00:00:00 2001 From: JorritvanGils Date: Wed, 10 Dec 2025 16:32:59 +0100 Subject: [PATCH] Use metagraph.n to solve testnet range mismatch --- neurons/validator.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/neurons/validator.py b/neurons/validator.py index 365e1e48..0039c11d 100644 --- a/neurons/validator.py +++ b/neurons/validator.py @@ -95,7 +95,17 @@ def __init__(self, config=None): self._load_gradient_compressors() if self.master: map_uid_to_peerid(self) - for i in range(256): + + if self.master: + metagraph_n_value = int(self.metagraph.n) + metagraph_n_tensor = torch.tensor([metagraph_n_value], dtype=torch.int32, device="cpu") + else: + metagraph_n_tensor = torch.tensor([0], dtype=torch.int32, device="cpu") + + dist.broadcast(metagraph_n_tensor, src=0, group=self.gloo_group) + self.metagraph_n = metagraph_n_tensor[0].item() + + for i in range(self.metagraph_n): self.logger.info(i) self.save_gradient(self.global_progress.epoch, i) if self.master: