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
vision
Commits
461b034c
Unverified
Commit
461b034c
authored
Apr 10, 2022
by
Yulv-git
Committed by
GitHub
Apr 10, 2022
Browse files
Fix typos. (#5796)
* Update _register_onnx_ops.py * Update misc.py
parent
f5afae50
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
torchvision/ops/_register_onnx_ops.py
torchvision/ops/_register_onnx_ops.py
+1
-1
torchvision/ops/misc.py
torchvision/ops/misc.py
+1
-1
No files found.
torchvision/ops/_register_onnx_ops.py
View file @
461b034c
...
@@ -38,7 +38,7 @@ def _register_custom_op():
...
@@ -38,7 +38,7 @@ def _register_custom_op():
# ONNX doesn't support negative sampling_ratio
# ONNX doesn't support negative sampling_ratio
if
sampling_ratio
<
0
:
if
sampling_ratio
<
0
:
warnings
.
warn
(
warnings
.
warn
(
"ONNX doesn't support negative sampling ratio, therefore is
is
set to 0 in order to be exported."
"ONNX doesn't support negative sampling ratio, therefore is set to 0 in order to be exported."
)
)
sampling_ratio
=
0
sampling_ratio
=
0
return
g
.
op
(
return
g
.
op
(
...
...
torchvision/ops/misc.py
View file @
461b034c
...
@@ -219,7 +219,7 @@ class Conv3dNormActivation(ConvNormActivation):
...
@@ -219,7 +219,7 @@ class Conv3dNormActivation(ConvNormActivation):
class
SqueezeExcitation
(
torch
.
nn
.
Module
):
class
SqueezeExcitation
(
torch
.
nn
.
Module
):
"""
"""
This block implements the Squeeze-and-Excitation block from https://arxiv.org/abs/1709.01507 (see Fig. 1).
This block implements the Squeeze-and-Excitation block from https://arxiv.org/abs/1709.01507 (see Fig. 1).
Parameters ``activation``, and ``scale_activation`` correspond to ``delta`` and ``sigma`` in
in
eq. 3.
Parameters ``activation``, and ``scale_activation`` correspond to ``delta`` and ``sigma`` in eq. 3.
Args:
Args:
input_channels (int): Number of channels in the input image
input_channels (int): Number of channels in the input image
...
...
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