Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
one
spconv
Commits
8da6f967
Commit
8da6f967
authored
Feb 08, 2019
by
traveller59
Browse files
fix #17
parent
20557e83
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
spconv/conv.py
spconv/conv.py
+2
-2
No files found.
spconv/conv.py
View file @
8da6f967
...
...
@@ -134,7 +134,7 @@ class SparseConvolution(SparseModule):
input
.
features
=
torch
.
mm
(
input
.
features
,
self
.
weight
.
view
(
self
.
in_channels
,
self
.
out_channels
))
if
self
.
bias
:
if
self
.
bias
is
not
None
:
input
.
features
+=
self
.
bias
return
input
datas
=
input
.
find_indice_pair
(
self
.
indice_key
)
...
...
@@ -165,7 +165,7 @@ class SparseConvolution(SparseModule):
self
.
weight
,
indice_pairs
.
to
(
device
),
indice_pair_num
,
outids
.
shape
[
0
])
if
self
.
bias
:
if
self
.
bias
is
not
None
:
out_features
+=
self
.
bias
out_tensor
=
spconv
.
SparseConvTensor
(
out_features
,
outids
,
out_spatial_shape
,
batch_size
)
...
...
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