oneflow-documentation
oneflow-documentation copied to clipboard
nn.Graph Turotial of Activation Checkpointing
接口:https://oneflow.readthedocs.io/en/master/graph.html#oneflow.nn.graph.block_config.BlockConfig.activation_checkpointing
参考的例子: https://github.com/Oneflow-Inc/oneflow/blob/master/python/oneflow/test/graph/test_graph_activation_checkpoint.py
参考背景资料:https://pytorch.org/docs/stable/checkpoint.html
明天能达到的标准,我们约定为How-To:
- 大致的概念介绍
- API使用演示的一个How-To的例子
达到的效果:给我们自己的算法同事一个最基本的教程,他们可以参考把功能启动起来;而不是接口使用都要口口相传;