Sfoglia il codice sorgente

ignore_stale_updates

justheuristic 3 anni fa
parent
commit
927e31f27d
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      hivemind/optim/collaborative.py

+ 1 - 1
hivemind/optim/collaborative.py

@@ -118,7 +118,7 @@ class CollaborativeOptimizer(DecentralizedOptimizerBase):
         metadata_expiration: float = 60.0,
         averaging_timeout: Optional[float] = None,
         load_state_timeout: float = 600.0,
-        staleness_timeout: float = 15.0,
+        staleness_timeout: float = 30.0,
         step_tolerance: int = 1,
         reuse_grad_buffers: bool = False,
         accumulate_grads_on: Optional[torch.device] = None,