| 123456789101112 |
- # 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"]
|
