Aleksandr Borzunov преди 3 години
родител
ревизия
da979f7079
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 2 2
      arguments.py

+ 2 - 2
arguments.py

@@ -67,10 +67,10 @@ class CollaborativeArguments:
         default=15.0, metadata={"help": "Averaging group will wait for stragglers for at most this many seconds"}
         default=15.0, metadata={"help": "Averaging group will wait for stragglers for at most this many seconds"}
     )
     )
     allreduce_timeout: float = field(
     allreduce_timeout: float = field(
-        default=120, metadata={"help": "Give up on a given all-reduce round after this many seconds"}
+        default=60, metadata={"help": "Give up on a given all-reduce round after this many seconds"}
     )
     )
     averaging_timeout: float = field(
     averaging_timeout: float = field(
-        default=120, metadata={"help": "Give up on averaging step after this many seconds"}
+        default=180, metadata={"help": "Give up on averaging step after this many seconds"}
     )
     )
     reuse_grad_buffers: bool = field(default=True, metadata={
     reuse_grad_buffers: bool = field(default=True, metadata={
         "help": "Whether or not to use model's .grad buffers for accumulating gradients across local steps. This "
         "help": "Whether or not to use model's .grad buffers for accumulating gradients across local steps. This "