|
@@ -53,7 +53,7 @@ class SharedBytes:
|
|
|
"""Create another shared byte value, represented as a scalar uint8 tensor"""
|
|
|
with cls._lock:
|
|
|
if cls._pid != os.getpid() or cls._buffer is None or cls._index >= len(cls._buffer):
|
|
|
- buffer_size = int(os.environ.get("HIVEMIND_SHM_BUFFER_SIZE", 4096))
|
|
|
+ buffer_size = int(os.environ.get("HIVEMIND_SHM_BUFFER_SIZE", 16))
|
|
|
cls._pid = os.getpid()
|
|
|
cls._buffer = torch.empty([buffer_size], dtype=torch.uint8).share_memory_()
|
|
|
cls._index = 0
|