Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Stephan Seitz
pystencil_reco
Commits
585ac16b
Commit
585ac16b
authored
Feb 28, 2020
by
Stephan Seitz
Browse files
Call super resultion kernels
parent
9a53b949
Pipeline
#22366
failed with stage
in 1 minute and 18 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
tests/test_superresolution.py
View file @
585ac16b
...
...
@@ -67,7 +67,18 @@ def test_torch_simple():
x
=
torch
.
ones
((
10
,
40
)).
cuda
()
h
=
torch
.
ones
((
10
,
40
,
8
)).
cuda
()
kernel
().
forward
(
h
,
x
)
y
=
kernel
().
forward
(
h
,
x
)
# with autograd
x
=
torch
.
ones
((
10
,
40
),
requires_grad
=
True
).
cuda
()
h
=
torch
.
ones
((
10
,
40
,
8
),
requires_grad
=
True
).
cuda
()
y
=
kernel
().
forward
(
h
,
x
)[
0
]
assert
y
.
requires_grad
loss
=
y
.
mean
()
loss
.
backward
()
# kernel().forward(*([1]*9), x, y)
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new 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