瀏覽代碼

tweak args

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"}
     )
     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(
-        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={
         "help": "Whether or not to use model's .grad buffers for accumulating gradients across local steps. This "