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.

Sign up for GitHub

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

Jump to bottom

UNET self.ups alternative in forward pass #96

Open

gaffinetB wants to merge 1 commit into aladdinpersson:master from gaffinetB:master

gaffinetB commented Dec 30, 2021

Implemented alternative way to make skip connections work in forward pass for UNET model.

Instead of iterating with a step of 2 over self.ups, iterate over the reversed(skip_connections).
The indices are adapted accordingly.

Changes come as a reaction to PyTorch Image Segmentation Tutorial with U-NET: everything from scratch baby at 16'00'' (love the video)

Hope it helps. Hope it's more intuitive as approach.


          UNET self.ups alternative

afd3f48

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment