Set default --mean_balance_check_period = 60

pull/84/head
Aleksandr Borzunov 2 years ago
parent a7e2c2b5d8
commit 7c76610416

@ -87,7 +87,7 @@ def main():
help="Rebalance the swarm if its throughput is worse than this share of the optimal "
"throughput. Use 0.0 to disable rebalancing, values > 1.0 to force rebalancing "
"on each check for debugging purposes.")
parser.add_argument("--mean_balance_check_period", type=float, default=150,
parser.add_argument("--mean_balance_check_period", type=float, default=60,
help="Check the swarm's balance every N seconds (and rebalance it if necessary)")
parser.add_argument("--use_auth_token", type=str, default=None, help="auth token for from_pretrained")

@ -62,7 +62,7 @@ class Server(threading.Thread):
prefetch_batches: int = 1,
sender_threads: int = 1,
balance_quality: float = 0.75,
mean_balance_check_period: float = 150,
mean_balance_check_period: float = 60,
mean_block_selection_delay: float = 0.5,
use_auth_token: Optional[str] = None,
load_in_8bit: bool = False,

Loading…
Cancel
Save