Name	Size	Mode	Actions
skip/	-	0755	rm
_balance/	-	0755	rm
__pycache__/	-	0755	rm
batchnorm.py	5603	0644	editdlrm
checkpoint.py	11464	0644	editdlrm
copy.py	3716	0644	editdlrm
dependency.py	1694	0644	editdlrm
microbatch.py	7427	0644	editdlrm
phony.py	1536	0644	editdlrm
pipe.py	17883	0644	editdlrm
pipeline.py	9638	0644	editdlrm
stream.py	3807	0644	editdlrm
utils.py	1147	0644	editdlrm
worker.py	4284	0644	editdlrm
__init__.py	451	0644	editdlrm

# Copyright 2019 Kakao Brain # # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved. # # This source code is licensed under the BSD license found in the # LICENSE file in the root directory of this source tree. """A Pipe implementation in PyTorch.""" from .checkpoint import is_checkpointing, is_recomputing from .pipe import Pipe, WithDevice from .microbatch import NoChunk __all__ = ["Pipe", "is_checkpointing", "is_recomputing"]