Skip to content

GitLab

"megatron/model/gpt_model.py" did not exist on "456f17280fcc25eb6bb3d9de7f9cad170b7b98d9"
Switch branch/tag
Find file Normal viewHistoryPermalink
apis.rst 656 Bytes
Newer Older
unknown's avatar
添加mmclassification-0.24.1代码，删除mmclassification-speed-benchmark  
unknown 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 
.. role:: hidden
    :class: hidden-section

mmcls.apis
===================================

These are some high-level APIs for classification tasks.

.. contents:: mmcls.apis
   :depth: 2
   :local:
   :backlinks: top

.. currentmodule:: mmcls.apis

Train
------------------

.. autosummary::
   :toctree: generated
   :nosignatures:

   init_random_seed
   set_random_seed
   train_model

Test
------------------
.. autosummary::
   :toctree: generated
   :nosignatures:

   single_gpu_test
   multi_gpu_test

Inference
------------------

.. autosummary::
   :toctree: generated
   :nosignatures:

   init_model
   inference_model
   show_result_pyplot