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
torch-spline-conv
Commits
4cb72f68
"torchvision/vscode:/vscode.git/clone" did not exist on "dd66a9d83d499788a7589cc4bf0d5fb6bc3a2127"
Commit
4cb72f68
authored
Mar 13, 2018
by
Jan Eric Lenssen
Browse files
bugfix + faust script num features
parent
91ea59fc
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
2 deletions
+5
-2
spline_conv_gpu.py
spline_conv_gpu.py
+5
-2
No files found.
spline_conv_gpu.py
View file @
4cb72f68
...
...
@@ -449,9 +449,9 @@ def get_basis_kernel(k_max, K, dim, degree, dtype='float'):
def
get_basis_backward_kernel
(
k_max
,
K
,
dim
,
degree
,
dtype
=
'float'
):
if
degree
==
3
:
raise
NotImplementedError
_spline_kernel
=
_spline_kernel_linear_backward2
elif
degree
==
2
:
raise
NotImplementedError
_spline_kernel
=
_spline_kernel_linear_backward2
else
:
_spline_kernel
=
_spline_kernel_linear_backward2
...
...
@@ -548,6 +548,9 @@ class SplineConvGPU(Function):
num_threads
=
grad_output
.
numel
()
if
self
.
bp_to_adj
:
if
self
.
degree
==
2
or
self
.
degree
==
3
:
print
(
'Backward to u for degree>1 not implemented!'
)
raise
NotImplementedError
input
,
weight
,
adj_values
=
self
.
saved_tensors
#adj_values = torch.clamp(adj_values,min=0.0,max=1.0)
amount
=
self
.
amount
...
...
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