Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ApiReference.rst 2.5 KB
Newer Older
qianyj's avatar
the source code of NNI for DCU  
qianyj 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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123 
Retiarii API Reference
======================

.. contents::

Inline Mutation APIs
--------------------

..  autoclass:: nni.retiarii.nn.pytorch.LayerChoice
    :members:

..  autoclass:: nni.retiarii.nn.pytorch.InputChoice
    :members:

..  autoclass:: nni.retiarii.nn.pytorch.ValueChoice
    :members:

..  autoclass:: nni.retiarii.nn.pytorch.ChosenInputs
    :members:

..  autoclass:: nni.retiarii.nn.pytorch.Repeat
    :members:

..  autoclass:: nni.retiarii.nn.pytorch.Cell
    :members:

Graph Mutation APIs
-------------------

..  autoclass:: nni.retiarii.Mutator
    :members:

..  autoclass:: nni.retiarii.Model
    :members:

..  autoclass:: nni.retiarii.Graph
    :members:

..  autoclass:: nni.retiarii.Node
    :members:

..  autoclass:: nni.retiarii.Edge
    :members:

..  autoclass:: nni.retiarii.Operation
    :members:

Evaluators
----------

..  autoclass:: nni.retiarii.evaluator.FunctionalEvaluator
    :members:

..  autoclass:: nni.retiarii.evaluator.pytorch.lightning.LightningModule
    :members:

..  autoclass:: nni.retiarii.evaluator.pytorch.lightning.Classification
    :members:

..  autoclass:: nni.retiarii.evaluator.pytorch.lightning.Regression
    :members:

Oneshot Trainers
----------------

..  autoclass:: nni.retiarii.oneshot.pytorch.DartsTrainer
    :members:

..  autoclass:: nni.retiarii.oneshot.pytorch.EnasTrainer
    :members:

..  autoclass:: nni.retiarii.oneshot.pytorch.ProxylessTrainer
    :members:

..  autoclass:: nni.retiarii.oneshot.pytorch.SinglePathTrainer
    :members:

Exploration Strategies
----------------------

..  autoclass:: nni.retiarii.strategy.Random
    :members:

..  autoclass:: nni.retiarii.strategy.GridSearch
    :members:

..  autoclass:: nni.retiarii.strategy.RegularizedEvolution
    :members:

..  autoclass:: nni.retiarii.strategy.TPEStrategy
    :members:

..  autoclass:: nni.retiarii.strategy.PolicyBasedRL
    :members:

Retiarii Experiments
--------------------

..  autoclass:: nni.retiarii.experiment.pytorch.RetiariiExperiment
    :members:

..  autoclass:: nni.retiarii.experiment.pytorch.RetiariiExeConfig
    :members:

CGO Execution
-------------

..  autofunction:: nni.retiarii.evaluator.pytorch.cgo.evaluator.MultiModelSupervisedLearningModule

..  autofunction:: nni.retiarii.evaluator.pytorch.cgo.evaluator.Classification

..  autofunction:: nni.retiarii.evaluator.pytorch.cgo.evaluator.Regression

Utilities
---------

..  autofunction:: nni.retiarii.basic_unit

..  autofunction:: nni.retiarii.model_wrapper

..  autofunction:: nni.retiarii.fixed_arch