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
de879592
Commit
de879592
authored
Aug 07, 2018
by
rusty1s
Browse files
0.4.1
parent
e67711ee
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
7 additions
and
8 deletions
+7
-8
.coveragerc
.coveragerc
+2
-2
.travis.yml
.travis.yml
+3
-4
test/utils.py
test/utils.py
+1
-1
torch_spline_conv/weighting.py
torch_spline_conv/weighting.py
+1
-1
No files found.
.coveragerc
View file @
de879592
[report]
exclude_lines =
pragma: no cover
def backward
def bw
cuda
backward
.travis.yml
View file @
de879592
...
...
@@ -7,12 +7,11 @@ matrix:
-
python
:
3.5
-
python
:
3.6
install
:
-
if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.
0
-cp27-cp27mu-linux_x86_64.whl; fi
-
if [[ $TRAVIS_PYTHON_VERSION == 3.5 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.
0
-cp35-cp35m-linux_x86_64.whl; fi
-
if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.
0
-cp36-cp36m-linux_x86_64.whl; fi
-
if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.
1
-cp27-cp27mu-linux_x86_64.whl; fi
-
if [[ $TRAVIS_PYTHON_VERSION == 3.5 ]]; then pip install
http://download.pytorch.org/whl/cpu/torch-0.4.
1
-cp35-cp35m-linux_x86_64.whl; fi
-
if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then pip install http://download.pytorch.org/whl/cpu/torch-0.4.
1
-cp36-cp36m-linux_x86_64.whl; fi
-
pip install pycodestyle
-
pip install flake8
-
pip install cffi
-
pip install codecov
script
:
-
pycodestyle .
...
...
test/utils.py
View file @
de879592
...
...
@@ -3,7 +3,7 @@ import torch
dtypes
=
[
torch
.
float
,
torch
.
double
]
devices
=
[
torch
.
device
(
'cpu'
)]
if
torch
.
cuda
.
is_available
():
# pragma: no cover
if
torch
.
cuda
.
is_available
():
devices
+=
[
torch
.
device
(
'cuda:{}'
.
format
(
torch
.
cuda
.
current_device
()))]
...
...
torch_spline_conv/weighting.py
View file @
de879592
...
...
@@ -35,7 +35,7 @@ class SplineWeighting(Function):
return
fw
(
src
,
weight
,
basis
,
weight_index
)
@
staticmethod
def
backward
(
ctx
,
grad_out
):
# pragma: no cover
def
backward
(
ctx
,
grad_out
):
grad_src
=
grad_weight
=
grad_basis
=
None
src
,
weight
,
basis
,
weight_index
=
ctx
.
saved_tensors
...
...
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