def __init__(self, context=None, reducers=None):
if context is None:
context = multiprocessing
if reducers is None:
reducers = {}
for t in torch._tensor_classes:
reducers.setdefault(t, reduce_tensor)
for t in torch._storage_classes:
reducers.setdefault(t, reduce_storage)
super(Queue, self).__init__(context, reducers)
评论列表
文章目录