Skip to content

Commit 1a1dc5c

Browse files
author
Max Freudenberg
committed
add comments
1 parent 8b8a008 commit 1a1dc5c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ext/NNlibCUDA/src/upsample.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,8 @@
4242

4343
# Forward and backward pass have been tested to produce the same output
4444
# as pytorch with align_corners=True - it works modulo bit noise.
45+
# pytorch's default is align_corners=False, because otherwise the gradients depend on the
46+
# image size, which should be avoided -> this should be considered here as well
4547

4648
@inline function compute_source_index(ratio::T, dst_index, align_corners) where T
4749
if align_corners

0 commit comments

Comments
 (0)