瀏覽代碼

1mb parts

justheuristic 3 年之前
父節點
當前提交
4b1b57f0d0
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      hivemind/averaging/partition.py

+ 1 - 1
hivemind/averaging/partition.py

@@ -13,7 +13,7 @@ from hivemind.proto import runtime_pb2
 from hivemind.utils.asyncio import amap_in_executor
 
 T = TypeVar("T")
-DEFAULT_PART_SIZE_BYTES = 2 ** 16
+DEFAULT_PART_SIZE_BYTES = 2 ** 20
 
 
 class TensorPartContainer: