Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

introduce trilinear upsampling #5

Closed
wants to merge 70 commits into from
Closed

Conversation

maxfreu
Copy link
Contributor

@maxfreu maxfreu commented Apr 25, 2021

Hi! This is the counterpart to FluxML/NNlib.jl#315 The checks fail because the NNlib PR is not yet merged.

@DhairyaLGandhi
Copy link
Member

We should also discuss supporting interpolations.jl directly. They have added the plumbing required to make it compatible with zygote. We would still need to support getting the adjoints to the Interpolations themselves, but this is a promising move nonetheless.

maleadt and others added 4 commits April 29, 2021 11:56
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
@maxfreu maxfreu marked this pull request as ready for review April 30, 2021 11:53
@maxfreu maxfreu closed this May 5, 2021
@maxfreu maxfreu reopened this May 5, 2021
@maxfreu
Copy link
Contributor Author

maxfreu commented May 5, 2021

Maybe you can re-trigger the checks, as the corresponding NNlib PR has been merged.

@maxfreu
Copy link
Contributor Author

maxfreu commented May 19, 2021

@CarloLucibello @DhairyaLGandhi
I know it's not the best code in history, but I think after re-running the checks this is mergeable.

Anton Smirnov and others added 3 commits May 21, 2021 01:14
Update src/gather.jl

Co-authored-by: Carlo Lucibello <carlo.lucibello@gmail.com>
replace _check_dims as gather_check_dims

fix

update manifest.toml
ToucheSir and others added 28 commits June 27, 2021 07:59
Co-authored-by: Michael Abbott <32575566+mcabbott@users.noreply.github.com>
Test `softmax` with varying dimensions
add count_indices for cuarray

add CUDA kernel for divide_by_counts!

add NNlib.∇scatter_src for cuda gradient

support scatter mean AD for CUDA

support scatter *,/ AD for CUDA
Support scatter for CUDA gradient
Bump patch and check for failures with Julia 1.7-nightly
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
Add group support for convolutions
@maxfreu maxfreu closed this Jul 24, 2021
@maxfreu
Copy link
Contributor Author

maxfreu commented Jul 24, 2021

I made some git chaos. Will make a new PR.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.