Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
libai.utils.rst 1.23 KB
Newer Older
yuguo960516's avatar
bloom  
yuguo960516 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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54 
libai.utils
##############################

libai.utils.distributed module
---------------------------------
.. currentmodule:: libai.utils
.. automodule:: libai.utils.distributed
    :members:
        get_dist_util,
        setup_dist_util,
        ttol,
        tton,
        synchronize,
        convert_to_distributed_default_setting,
        get_nd_sbp,
        get_layer_placement,
        get_world_size,
        get_num_nodes,
        get_rank,
        get_local_rank,
        same_sbp,
        get_data_parallel_size,
        get_data_parallel_rank,
        get_hidden_sbp
        
libai.utils.events module
---------------------------------
.. currentmodule:: libai.utils
.. automodule:: libai.utils.events
    :members:
        get_event_storage,
        JSONWriter,
        CommonMetricPrinter,
        EventStorage,


libai.utils.logger module
---------------------------------
.. currentmodule:: libai.utils
.. automodule:: libai.utils.logger
    :members:
        setup_logger,
        log_first_n,
        log_every_n,
        log_every_n_seconds


libai.utils.checkpoint module
---------------------------------
.. currentmodule:: libai.utils
.. automodule:: libai.utils.checkpoint
    :members:
        Checkpointer,
        PeriodicCheckpointer