libai.engine.rst 861 Bytes
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
libai.engine
##############################

libai.engine.default module
---------------------------------
.. currentmodule:: libai.engine
.. automodule:: libai.engine.default
    :member-order: bysource
    :members: 
        default_setup,
        DefaultTrainer,


libai.engine.hooks module
---------------------------------
.. currentmodule:: libai.engine
.. automodule:: libai.engine.hooks
    :member-order: bysource
    :members: 
        CallbackHook,
        IterationTimer,
        PeriodicWriter,
        PeriodicCheckpointer,
        BestCheckpointer,
        EvalHook,
        LRScheduler,


libai.engine.trainer module
---------------------------------
.. currentmodule:: libai.engine
.. automodule:: libai.engine.trainer
    :member-order: bysource
    :members: 
        HookBase,
        TrainerBase,
        EagerTrainer,
        GraphTrainer,