oneflow-documentation icon indicating copy to clipboard operation
oneflow-documentation copied to clipboard

nn.Graph Turotial of Activation Checkpointing

Open strint opened this issue 3 years ago • 1 comments

接口: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

strint avatar Mar 17 '22 09:03 strint

明天能达到的标准,我们约定为How-To:

  • 大致的概念介绍
  • API使用演示的一个How-To的例子

达到的效果:给我们自己的算法同事一个最基本的教程,他们可以参考把功能启动起来;而不是接口使用都要口口相传;

strint avatar Mar 17 '22 10:03 strint