Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
mmdetection3d
Commits
29b546d0
Commit
29b546d0
authored
Apr 20, 2020
by
zhangwenwei
Browse files
Add numba requirements and doc
parent
6d71b439
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
17 additions
and
22 deletions
+17
-22
.travis.yml
.travis.yml
+3
-3
mmdet3d/core/optimizer/cocktail_constructor.py
mmdet3d/core/optimizer/cocktail_constructor.py
+13
-19
requirements/runtime.txt
requirements/runtime.txt
+1
-0
No files found.
.travis.yml
View file @
29b546d0
dist
:
bionic
# ubuntu 18.04
dist
:
bionic
# ubuntu 18.04
language
:
python
language
:
python
python
:
python
:
...
@@ -24,8 +24,8 @@ before_install:
...
@@ -24,8 +24,8 @@ before_install:
-
PATH=${CUDA_HOME}/bin:${PATH}
-
PATH=${CUDA_HOME}/bin:${PATH}
install
:
install
:
-
pip install Pillow==6.2.2
# remove this line when torchvision>=0.5
-
pip install Pillow==6.2.2
# remove this line when torchvision>=0.5
-
pip install torch==1.2 torchvision==0.4.0
# TODO: fix CI for pytorch>1.2
-
pip install torch==1.2 torchvision==0.4.0
# TODO: fix CI for pytorch>1.2
-
pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"
-
pip install "git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI"
-
pip install -r requirements.txt
-
pip install -r requirements.txt
...
...
mmdet3d/core/optimizer/cocktail_constructor.py
View file @
29b546d0
...
@@ -7,28 +7,22 @@ from .cocktail_optimizer import CocktailOptimizer
...
@@ -7,28 +7,22 @@ from .cocktail_optimizer import CocktailOptimizer
@
OPTIMIZER_BUILDERS
.
register_module
@
OPTIMIZER_BUILDERS
.
register_module
class
CocktailOptimizerConstructor
(
object
):
class
CocktailOptimizerConstructor
(
object
):
"""Default constructor for optimizers.
"""Special constructor for cocktail optimizers.
This constructor constructs cocktail optimizer for multi-modality
detectors. It builds separate optimizers for separate branchs for
different modalities. More details can be found in the ECCV submission
(to be release).
Attributes:
Attributes:
model (:obj:`nn.Module`): The model with parameters to be optimized.
model (:obj:`nn.Module`): The model with parameters to be optimized.
optimizer_cfg (dict): The config dict of the optimizer.
optimizer_cfg (dict): The config dict of the optimizer. The keys of
Positional fields are:
the dict are used to search for the corresponding keys in the
- type: class name of the optimizer.
model, and the value if a dict that really defines the optimizer.
- lr: base learning rate.
See example below for the usage.
Optional fields are:
paramwise_cfg (dict): The dict for paramwise options. This is not
- any arguments of the corresponding optimizer type, e.g.,
supported in the current version. But it should be supported in
weight_decay, momentum, etc.
the future release.
paramwise_cfg (dict, optional): Parameter-wise options. Accepted fields
are:
- bias_lr_mult: It will be multiplied to the learning rate for
all bias parameters (except for those in normalization layers).
- bias_decay_mult: It will be multiplied to the weight decay for
all bias parameters (except for those in normalization layers and
depthwise conv layers).
- norm_decay_mult: will be multiplied to the weight decay
for all weight and bias parameters of normalization layers.
- dwconv_decay_mult: will be multiplied to the weight decay
for all weight and bias parameters of depthwise conv layers.
Example:
Example:
>>> import torch
>>> import torch
...
...
requirements/runtime.txt
View file @
29b546d0
matplotlib
matplotlib
mmcv>=0.2.15
mmcv>=0.2.15
numba>=0.45.1
numpy
numpy
# need older pillow until torchvision is fixed
# need older pillow until torchvision is fixed
Pillow<=6.2.2
Pillow<=6.2.2
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment