Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
graph.rst 1.06 KB
Newer Older
yuguo's avatar
Oneflow 0.8 for DCU  
yuguo committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39 
oneflow.nn.Graph
============================================================
Base class for running neural networks in Static Graph Mode.
------------------------------------------------------------
.. currentmodule:: oneflow.nn
.. autoclass:: oneflow.nn.Graph
    :members: __init__,
            build,
            __call__,
            add_optimizer,
            set_grad_scaler,
            state_dict,
            load_state_dict,
            name,
            debug,
            __repr__,
    :member-order: bysource



.. autoclass:: oneflow.nn.graph.graph_config.GraphConfig
    :members: enable_amp,
            enable_zero,
            allow_fuse_model_update_ops,
            allow_fuse_add_to_output,
            allow_fuse_cast_scale,
            set_gradient_accumulation_steps,
            enable_cudnn_conv_heuristic_search_algo,
            enable_straighten_algorithm,
    :member-order: bysource



.. autoclass:: oneflow.nn.graph.block_config.BlockConfig
    :members: stage_id,
            set_stage,
            activation_checkpointing,
    :member-order: bysource