Skip to content

GitLab

Unverified Commit 535eb011 authored by Min Xu's avatar Min Xu Committed by GitHub
Browse files

[fix] expose checkpoint_wrapper (#399) 

* [fix] expose checkpoint_wrapper

* fix formatting
parent 205af8c2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
fairscale/nn/misc/__init__.py
View file @ 535eb011
......@@ -3,4 +3,5 @@
# This source code is licensed under the BSD license found in the
# LICENSE file in the root directory of this source tree.
from .checkpoint_activations import checkpoint_wrapper
from .flatten_params_wrapper import FlattenParamsWrapper
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment