Machine-Learning-Collection icon indicating copy to clipboard operation
Machine-Learning-Collection copied to clipboard

UNET self.ups alternative in forward pass

Open gaffinetB opened this issue 3 years ago • 0 comments

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.

gaffinetB avatar Dec 30 '21 17:12 gaffinetB